参考网址: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相关的错误。
训练模型和测试之前要做到心中有数,明确步骤,知道每一步为什么做,
硬件要求和软件要求都得提前考虑。
原文地址:https://www.cnblogs.com/gloria11/p/9799928.html
时间: 2024-10-16 19:47:59