- PyTorch == 1.0.0
- TorchVision == 0.2.1
- CUDA 10.0 cudnn 7.2
Please download backbone pretrained model from https://1drv.ms/u/s!Aus8VCZ_C_33dKvqI6pBZlifgJk, and put it into backbone package.
Please Run train.py
Please Run infer.py
Please alter path in config.py and datasets.py