/PyTorch-SPNet

Enhanced One-Stage Network

Primary LanguagePython

PyTorch-SPNet

Enhanced One-Stage Network

Required: pytorch: 0.4.0 anaconda: 3.6.0

Download COCO dataset: bash get_coco_dataset.sh

Testing: You need to rectify the directory(cause that I used many absolute path) in the test_images.py

Training: You need to rectify the directory(cause that I used many absolute path) in the training.py