Caffe:画CNN

2024-10-03 11:26:07 发布

您现在位置:Python中文网/ 问答频道 /正文

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

 python draw_net.py test.protxt test.png

它画不出来。它不会显示任何错误,但会显示结果测试.png文件是白色空白图像文件。谁能帮我修一下吗?这将有助于快速设计新的网络。在


Tags: 文件代码pytest网络net定义png
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

我也有同样的问题。基于this thread,我已经按照建议使用旧的Proto语法来解决这个问题。例如,我必须这样做:

将层定义从layers重命名为layer。通过caffe文档(或示例原型文件)重命名所有层类型-即层type: CONVOLUTION到{}等。替换较新的语法:

blobs_lr: 1         
blobs_lr: 1         
weight_decay: 1
weight_decay: 0

^{pr2}$

现在解析和新绘图工作得很好。参考caffe包中的example.prototxt文件以获得更好的直观性,proto语法是如何工作的。在

相关问题 更多 >