Caffe draw_net.py无法从原型文件中提取VGG-16

Question

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

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
  }
}