- Install dependencies
pip install -r requirements.txt
- Download data
bash scripts/download_data.sh
python train.py --imgs_dir [] --masks_dir []
python test.py --imgs_dir [] --masks_dir [] --model_name []
python demo.py --imgs_dir [] --result_dir [] --model_name []
- Reduce batch size to avoid CUDA out of memory.