Install Pytorch-1.1

$ conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

Configure the data file and output paths in config.json

install all the requirements in requirements.txt file
$ export CUDA_VISIBLE_DEVICES=0,1,2,3
$ python main.py

Name		Name	Last commit message	Last commit date
Latest commit History 36 Commits
Synchronized		Synchronized
cc_attention		cc_attention
dataset		dataset
loss		loss
networks		networks
utils		utils
.gitignore		.gitignore
README.md		README.md
config.json		config.json
config_server2.json		config_server2.json
engine.py		engine.py
evaluate.py		evaluate.py
main.py		main.py
metric.py		metric.py
myutils.py		myutils.py
requirements.txt		requirements.txt
scripts.txt		scripts.txt
seperating_images_script.py		seperating_images_script.py
train.py		train.py
write.py		write.py