Artificial intelligence 将模型从Caffe转换为ONNX
我想用Javascript制作原型,用一个在计算机上训练过的模型来探测天空 我尝试使用将Caffe模型(上面的prototxt和经过训练的数据)转换为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
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
}
}
...
我是一个新手,所以我可能会错过一些基本的东西,但是如果你有过类似的经历并且能够解决它,你能分享一下吗