search

htshinichi / caffe-onnx

caffe model convert to onnx model
MIT License
175 stars 43 forks source link

转换出现问题 #1

Closed zhangbohnu closed 5 years ago

zhangbohnu commented 5 years ago

您好， 用了您的代码依然无法转换 3.模型存在问题,未保存成功: Op registered for Upsample is depracted in domain_version of 10

==> Context: Bad node spec: input: "layer85-act_Y" input: "layer86-upsample_Scale" output: "layer86-upsample_Y" name: "layer86-upsample" op_type: "Upsample" attribute { name: "mode" s: "nearest" type: STRING }

ZQPei commented 5 years ago

是这样的，onnx0.5之后，ir_version为10，从这个版本开始，Upsample变为了Resize。 所以你有以下两种方法：

  1. 修改源码，把Upsample改为Resize
  2. 把onnx换成0.4的版本
gttiankai commented 5 years ago

@ZQPei 能详细说下怎么修改源码吗? 是将 caffe-onnx 中的所有的Upsample 修改为 Resize? 还是说修改需要转换的 prototxt?

gttiankai commented 5 years ago

@zhangbohnu 最终你是怎么修改的啊?