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

参考网址: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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值