This repository contains implementations of DCSNet in PyTorch. More information can be found here.
To use this package, install the required python packages (tested with python 3.6 on Ubuntu 16.04LTS):
pip install -r requirements.txt
Finally, to launch a training session, got to the models folder and run the following script:
python3 train.py --data-path <path to dataset folder> --exp-dir <path to summary folder> --device-num 0