Caffe：绘制CNN网

Question

我使用python代码绘制在prototext文件中定义的Net：

 python draw_net.py test.protxt test.png

无法画画。它没有显示任何错误，但结果test.png文件是白色空白图像文件。任何人都可以帮我修理它吗？这对于快速设计新网真的很有帮助。

Answer 1

我有同样的问题。基于this thread，我已设法通过使用较旧的Proto语法来解决此问题。例如，我必须这样做：

将图层定义从layer重命名为type: CONVOLUTION。所有图层类型都通过caffe文档（或示例原型文件）重命名 - 即图层type: "Convolution"到blobs_lr: 1 blobs_lr: 1 weight_decay: 1 weight_decay: 0 等。替换更新的语法：

param {
    name: "conv1_w"
    lr_mult: 1              
    decay_mult: 1
}
param {
    name: "conv1_b"
    lr_mult: 2              
    decay_mult: 0
}

代表

{{1}}

现在解析和新绘图效果很好。请参阅caffe包中的示例.prototxt文件以获得更好的直觉，工作原型语法如何。

Answer 2

2014年年中，Caffe changed their proto definition for extensibility造成了这个问题。作为此更改的结果，所有proto文件都必须更新为更新的定义。

为此，Caffe在distribute/bin/或.build_release/tools目录中提供以下工具：

1. upgrade_net_proto_binary.bin
2. upgrade_net_proto_text.bin

以下是如何将原型文本文件转换为更新格式的简单示例：

./upgrade_net_proto_text.bin /path/to/older_proto_file /path/to/newer_ouput_proto_file