用yolov3训练自己数据的一些坑的总结

  1. 用pytorch实现yolov3算法不是特别合适,目前网上的一些yolov3 pytorch复现经过我的亲身经历,感觉都有些问题。主要是test.py中的evaluate函数在yolo检测不到框的时候就会报错,我看了github上源代码的issues,认为解决方法就是如果报错就跳过evaluate,但我自己没有试过,以后有机会再添这坑。
  2. 目前我用过的比较好的yolov3复现是在这里:
git clone https://github.com/pjreddie/darknet

可以参考这两个博客
https://blog.csdn.net/qq_35451572/article/details/80384674
https://blog.csdn.net/lilai619/article/details/79695109
里面讲的比较详细了,我这就对其中一些小坑进行总计。
3. 自己上手不要用COCO VOC这些数据集,训练要很久才有效果。
4. 训练中出现nan是正常的,前期检测不到就会出现nan的
5. 如果要用gpu要装cuda和cudn,并且配置好。
6. 训练或者测试前要注意cfg有没有修改好相应的batchsize。
7. make -j 出现警告C4996 ‘cudaThreadSynchronize’: 被声明为已否决
参考这个:
https://blog.csdn.net/qq_42403190/article/details/90109282

  • 1
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 13
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 13
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值