python - Caffe:绘制 CNN 网络

Question

我使用 python 代码绘制了 prototext 文件中定义的网络:

 python draw_net.py test.protxt test.png

绘制失败。它没有显示任何错误，但结果 test.png 文件是白色空白图像文件。谁能帮我解决这个问题？这确实有助于快速设计新网络。

Answer 1

我遇到了同样的问题。基于this thread ，我已经按照建议使用旧的 Proto 语法设法解决了这个问题。例如我必须这样做:

将层定义从 layers 重命名为 layer。通过 caffe 文档(或示例原型(prototype)文件)重命名所有层类型 - 即层 type: CONVOLUTION 到 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
}

现在解析和新绘图工作正常。请引用 caffe 包中的示例 .prototxt 文件以获得更好的直觉，即工作原型(prototype)语法的样子。