Artificial intelligence 将模型从Caffe转换为ONNX

Artificial intelligence 将模型从Caffe转换为ONNX,artificial-intelligence,caffe,onnx,Artificial Intelligence,Caffe,Onnx,我想用Javascript制作原型,用一个在计算机上训练过的模型来探测天空 我尝试使用将Caffe模型(上面的prototxt和经过训练的数据)转换为ONNX模型 执行以下命令: mmconvert --srcFramework caffe --inputWeight baseline.caffemodel --inputNetwork deploy.net --dstFramework onnx --outputModel baseline.onnx --inputShape 1 3 240

我想用Javascript制作原型,用一个在计算机上训练过的模型来探测天空

我尝试使用将Caffe模型(上面的prototxt和经过训练的数据)转换为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第一部分的摘录。标签_0连接到虚拟_数据层和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
  }
}
...
我是一个新手,所以我可能会错过一些基本的东西,但是如果你有过类似的经历并且能够解决它,你能分享一下吗