Caffe draw_net.py 无法从 prototxt 文件中绘制 VGG-16

Question

我正在使用以下命令从caffe中的prototxt文件绘制网络框图

python draw_net.py <filename.prototxt> <output.png>

如果我使用 Alexnet、BVLC Caffenet 甚至 RCNN，这会很好。但是当我使用 VGG-16 文件时，它会给出一个大小为 11x11 的空白输出图像。没有错误被抛出。我也验证了路径。所有文件均取自Caffe Model Zoo。我正在使用从 master branch 获取的 Caffe。

Answer 1

您的VGG16 文件可能包含层的旧类型定义：

layers {
  bottom: "data"
  top: "conv1_1"
  name: "conv1_1"
  type: CONVOLUTION
  convolution_param {
    num_output: 64
    pad: 1
    kernel_size: 3
  }
}

要使其工作，您需要使用 type 的新 API：

layer {
  bottom: "conv1_1"
  top: "conv1_2"
  name: "conv1_2"
  type: "Convolution"
  convolution_param {
    num_output: 64
    pad: 1
    kernel_size: 3
  }
  param {
    lr_mult: 0
  }
  param {
    lr_mult: 0
  }
}