首页 > 其他 > 详细

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

时间:2018-10-16 19:16:29      阅读:156      评论:0      收藏:0      [点我收藏+]

参考网址: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相关的错误。
训练模型和测试之前要做到心中有数,明确步骤,知道每一步为什么做,
硬件要求和软件要求都得提前考虑。

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

原文:https://www.cnblogs.com/gloria11/p/9799928.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!