Windows下利用Python3绘制Caffe中的网络结构

最新推荐文章于 2018-10-04 20:50:54 发布

原创

最新推荐文章于 2018-10-04 20:50:54 发布 · 445 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#Caffe #python3

在Windows10+Anaconda3环境下利用Caffe中的draw_net.py文件绘制Caffe中的网络结构图

1.必要安装

第一步：安装pydotplus。在cmd命令窗口输入：pip install pydotplus。
第二步：安装Graphviz-2.38。在http://www.graphviz.org/Download_windows.php页面中下载Graphviz安装包，并完成安装（例如安装目录：C:/Program Files (x86)/Graphviz2.38/bin/）；右击“此电脑”打开“属性”，进入系统变量，添加C:/Program Files (x86)/Graphviz2.38/bin/到path中。

2.修改draw_net.py和draw.py文件

（1）打开caffe的根目录下python/caffe/draw.py文件，修改get_layer_label函数中的elif layer.type == 'Pooling':以下的内容，修改如下：

def get_layer_label(layer, rankdir):
    """Define node label based on layer type.

    Parameters
    ----------
    layer : ?
    rankdir : {'LR', 'RL', 'TB', 'BT'}
        Direction of graph layout.

    Returns
    -------
    string :
        A label for the current layer
    """

    if rankdir in ('TB', 'BT'):
        # If graph orientation is vertical, horizontal space is free and