Environment setup:
- Python 3.7.3
- conda install -c conda-forge/label/cf202003 opencv
- pip install matplotlib
- pip install seaborn
- pip install scikit-learn
- conda install -c pytorch pytorch==1.3.1
- conda install -c pytorch torchvision==0.4.2
Download weights file and put inside the weights folder
Google Drive: Google Drive Download
Run following script to download the coco datasets(18.8 GB) in root directory, if you want to test the MAP:
YOLObile/data/get_coco2014.sh
Adjust the path in YOLObile\data\coco2017.data
run following command to test:
python test.py --img-size 320 --batch-size 64 --device 0 --cfg cfg/csdarknet53s-panet-spp.cfg --weights weights/best8x-514.pt --data data/coco2017.data