转换具有多个节点的 Caffe 层

Question

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

"Multiple top nodes are not supported" 

我在 prototxt 中确实有一些具有多个顶部节点的层：

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 中的 cmets）。

你的选择是：

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