ASFF icon indicating copy to clipboard operation
ASFF copied to clipboard

can't find anything on VOC dataset by demo.py

Open zhan8002 opened this issue 4 years ago • 1 comments

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?

zhan8002 avatar Dec 24 '19 13:12 zhan8002

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.

GOATmessi7 avatar Dec 25 '19 06:12 GOATmessi7