将模型从Caffe转换为ONNX

时间:2020-11-09 00:39:50

标签: artificial-intelligence caffe onnx

我想使用在SkyFinder dataset上训练的模型对Java进行原型制作,以检测天空。

我尝试使用here将发布的MMdnn的Caffe模型(上面的原型和训练数据)转换为ONNX模型。

执行以下命令

mmconvert --srcFramework caffe --inputWeight baseline.caffemodel --inputNetwork deploy.net --dstFramework onnx --outputModel baseline.onnx --inputShape 1 3 240 320

出现此错误

mmdnn.conversion.caffe.errors.ConversionError: Layer not found: label_0

下面是上面prototxt的第一部分的摘录。 label_0已连接到dummy_data和convine_1层,但似乎未在任何地方定义。

name: "DeepBlueSky"
input: "data"
input_dim: 1
input_dim: 3
input_dim: 240
input_dim: 320 
layers {
  name: "dummy_data"
  type: DUMMY_DATA
  top: "label_0"
  dummy_data_param {
    data_filler {
      type: "constant"
      std: .5 
    }
    num: 1 
    channels: 1
    height: 240
    width: 320
  }
}

layers {
  name: "combine_1"
  type: CONCAT
  bottom: "data"
  bottom: "label_0"
  top: "combine_1"
  concat_param {
    concat_dim: 1
  }
}
...

我是新手,所以可能我错过了基础知识,但是如果您有类似的经历并且能够解决它,请您分享一下?

0 个答案:

没有答案