命令行操作¶
执行训练、测试、检测命令如下
相关参数¶
- 数据集:
PASCAL VOC 07+12
- 基础网络:
VGG16
- 输入大小:300\times 300
GPU: Nvidia GTX 1080Ti
训练¶
# 单GPU训练
$ python train.py --config-file configs/vgg_ssd300_voc0712.yaml
测试¶
# 单GPU测试
python test.py --config-file configs/vgg_ssd300_voc0712.yaml
检测¶
$ python demo.py --config-file configs/vgg_ssd300_voc0712.yaml --images_dir demo --ckpt outputs/vgg_ssd300_voc0712/model_final.pth