ASFF
ASFF copied to clipboard
can't find anything on VOC dataset by demo.py
Hello, I have trained the weight by VOC2007 dataset, and get the evaluate result map = 40. But when I use the my weight to test by demo.py , I can't get any prediction. The training script is: python -m torch.distributed.launch --nproc_per_node=1 --master_port=${RANDOM+10000} main.py --cfg config/yolov3_baseline.cfg -d VOC --ngpu 1 --distributed --checkpoint weights/darknet53_feature_mx.pth --start_epoch 0 --half --asff --rfb --dropblock -s 608 and parameters in demo.py is: --img XX --checkpoint XX --asff --rfb --half -d VOC -s 608 and I try some your weights, but don't run with parameter -d VOC. So should I change some code in demo.py to fit VOC?
40 mAP on VOC is quite low actually. Our experiments on VOC get about 84 mAP on VOC2007. If you don't have multi-gpu resources, I suggest you use mobilenet backbone and set larger batchsize for single-gpu training.