我想使用在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
}
}
...
我是新手,所以可能我错过了基础知识,但是如果您有类似的经历并且能够解决它,请您分享一下?