Caffe:绘制CNN网

Question

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

 python draw_net.py test.protxt test.png

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

Answer 1

我有同样的问题。基于这个线程，我已经按照建议使用旧的 Proto 语法设法解决了这个问题。例如我必须这样做：

将图层定义从重命名layers为layer。所有图层类型均由 caffe 文档（或示例 proto 文件）重命名 - 即图层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 文件，以获得更好的直觉，了解工作原型语法的样子。