使用多个节点转换Caffe图层

Question

我正在使用caffe-tensorflow来转换模型。 我收到了错误：

"Multiple top nodes are not supported" 

我在原型文件中有一些层有多个顶级节点：

layer {
  name: "slice2"
  type: "Slice"
  bottom: "conv2"
  top: "slice2_1"
  top: "slice2_2"
  slice_param {
    slice_dim: 1
  }
}

我有办法进行转换吗？ （有或没有caffe-tensorflow）

谢谢

Answer 1

你不能在caffe-tensorflow中做到这一点，它只支持单个顶部（输出）节点（参见graph.py中的注释）。

您的替代方案是：

• 克隆github代码并添加功能
• 在Tensorflow代码中实现您的模型。