码迷,mamicode.com
首页 > 其他好文 > 详细

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

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

标签:1.5   ESS   终端   ==   com   检测   预训练模型   没有   求和   

参考网址: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物体检测模型训练和测试总结

标签:1.5   ESS   终端   ==   com   检测   预训练模型   没有   求和   

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

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!