README.md

Training

Environment

1. Install dependencies

   pip install -r requirements.txt

2. Download data

   bash scripts/download_data.sh

Usage

python train.py --imgs_dir [] --masks_dir []

Validate

python test.py --imgs_dir [] --masks_dir [] --model_name []

Testing

Usage

python demo.py --imgs_dir [] --result_dir [] --model_name []

• Reduce batch size to avoid CUDA out of memory.