yolov5转caffe过程中的各种坑!!从caffe编译到yolov5转onnx中后处理模块的处理!

1. 编译caffe的坑

重点!!anaconda与caffe编译会有一些冲突!建议先编译caffe的时候把conda虚拟环境关闭,最好是在没有conda的环境安装!!
如果安装了conda那一定一定要记得区分加sudo和不加sudo的区别!!!这在任何装有conda的虚拟环境下都适用。简单来说加了sudo等于使用默认的python环境!
编译caffe的时候一定一定注意区分

make all -j12
和
sudo make all -j12

另外要注意的是一定检查好各种路径还有名称,能避免很多错误。其余问题大多数网上都有解决方案。

2.pt->onnx

yolov5s转onnx的时候最好不要转后处理的部分,转caffe确实会有些问题。但是重点是能查到的解决方案里没有找到告诉你怎么转能去掉后处理的!大部分都是说加上参数 --train。但是新版本的yolov5(6.2+)是没有–train这个参数的!
最后的解决方案是加一行

training=torch.onnx.TrainingMode.TRAINING

如图:
在这里插入图片描述转caffe model的时候如果加了后处理转不过可以试一试这个方法(别的框架类似问题也可以用同样的思路尝试解决)

贴一张加了后处理的yolov5网络结构。
在这里插入图片描述

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值