linjohnss/DIP_Final

Water Segmentation

DIP_Final

Zero Shot Water Segmentation

Requirement

Install CLIP Seg

pip install torch torchvision
pip install -q transformers

Use fast bilatral solver in opencv

Use fast bilatral solver in opencv

Quick start

Run CLIP Seg

python dip_final.py

Run FBS

python fbs.py

Run on demo dataset

python run_demo.py