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网络结构。