RichardSong-dev
RichardSong-dev
> @MTlab I also met this problem, when convert pytorch yolov3 to onnx, then convert to caffe > 我也遇到这个问题,在用yolov3转换到onnx, 再转换到caffe 所以老哥问题解决了吗?我估计和你情况一样,pytorch->onnx->caffe。用opset9就报这个错,opset11在转upsample时报KeyError: 'height_scale'。
> onnx 的opset版本是9吗?或者你把模型给我,我给你看看。 opset应该是9,之前pytorch导出时没指定version应该是默认9。我另一个模型yolo在转换reshape节点时还出现了其他错误: Traceback (most recent call last): File "convertCaffe.py", line 122, in convertToCaffe(graph, opset_version, prototxt_path, caffemodel_path) File "convertCaffe.py", line 67, in convertToCaffe layer = converter_fn(node,graph,err) File "/root/ONNXToCaffe-master/onnx2caffe/_operators.py", line...
> @RichardSong-dev upsample的bug修复了,源码已更新,你试试看,然后你的模型里面的reshape操作维度大于4维了,这个caffe不支持,所以你导出onnx时,需要把这部分去掉才行。 > 还有就是你的caffe版本需要添加https://github.com/jnulzl/caffe_plus 这个项目中Upsample实现才能支持nearest模式的操作 谢谢大佬,我去研究研究!
@xxradon 大佬,昨天按照您的方法试了一下,已经可以转换成功了!但当我在pytorch中把upsample改为bilinear之后又出现了问题: Traceback (most recent call last): File "convertCaffe.py", line 122, in convertToCaffe(graph, opset_version, prototxt_path, caffemodel_path) File "convertCaffe.py", line 67, in convertToCaffe layer = converter_fn(node,graph,err) File "/root/ONNXToCaffe/onnx2caffe/_operators.py", line 402, in...
@xxradon 大佬,对于昨天的问题,我按照您的方式改了一下_operators.py代码: ```python elif str(mode,encoding="gbk") == "linear": scales = node.input_tensors.get(node.inputs[1]) scale = scales[2] layer = myf("Upsample", node_name, [input_name], [output_name], upsample_param=dict( scale = int(scale) )) ``` 然后,使用https://github.com/Royzon/caffe_extension_layer 这个项目中的upsample layer重新编译了一下caffe。模型可以转换成功,但显示model output different。cos...
@bestzsq 把https://github.com/jnulzl/caffe_plus 这个项目中 include/caffe/layers/upsample_layer.hpp src/caffe/layers/upsample_layer.cpp src/caffe/layers/upsample_layer.cu 这三个文件放到你caffe目录相应的位置下,然后更改src/caffe/proto/caffe.proto中的代码,根据注释在相应位置添加如下代码: ```cpp optional UpsampleParameter upsample_param = 151; //数值根据你使用的caffe.proto中注释更改 /* ... */ message UpsampleParameter { optional int32 height = 1 [default = 32]; optional int32...
> 你好,参考你的例子来做yolov3的剪枝,现在遇到上采样upsampling卡住了,不知大佬有什么建议吗 tensorflow上的yolov3剪枝大佬有什么进展么