eval_ssd.py

eval_ssd.md

• parse_args
  • data-shape: 默认是 300，这个数值在 Evaluation 中会影响 net_name 和 SSDDefaultValTransform 里面的 imresize 也就是 Eval 图像的大小
• get_dataset
• get_dataloader
• benchmarking
• validate
• __main__

Eval Script

$ python eval_tiny_ssd.py --network vgg16_atrous --data-shape 300 --dataset custom --num-gpus 2 

自己的 model

$ python eval_tiny_ssd.py --pretrained false --gpus 0,1 --split mini_test --nms-thresh 0.1