ssd物体检测模型训练和测试总结

Posted 2021-01-13 gloria11

参考网址：github:https://github.com/naisy/realtime_object_detection

2018.10.16
ssd物体检测总结：
切记粗略地看一遍备注就开始训练模型
出现的错误：1、用branch1.5,tensorflow-gpu==1.8训练的模型在GT730,显存2g,运行不了，tensorflow-gpu==1.5没有NoMaxSuppressionv3，
2、用预训练模型ssd_mobilenet_coco_2018_1_28,tensorflowgpu==1.5训练，模型在cuda9.0,cudnn7.05上训练，得到的模型在GT730上2fps,
3、用预训练模型ssd_mobilenet_coco_2017_11_27,tensorflow-gpu==1.5训练，模型在GT730上18fps,
4、同时开两个终端，7fps,三个终端4fps,

5、选择nms_v1还是nms_v2,在测试备注里有说明，
6、模型一部分用cudnn7.1训练，一部分用cudnn7.05训练会报cudnn相关的错误。
训练模型和测试之前要做到心中有数，明确步骤，知道每一步为什么做，
硬件要求和软件要求都得提前考虑。