python画神经网络结构图_神经网络结构画图

参考代码：https://github.com/gwding/draw_convnet

环境

本地：win10+python3+tensorflow 0.12rc

远程ssh：Ubuntu14+python2,3+tensorflow 0.12rc

代码作者推荐运行如下

python3 -m venv .venv

source .venv/bin/activate

pip install -r requirements_python3.txt

python3 draw_convnet.py

本地远程均无法运行

远程ssh，远程因没有图形化的界面，提示：_tkinter.TclError: no display name and no $DISPLAY environment variable

经查找，参考https://stackoverflow.com/questions/4931376/generating-matplotlib-graphs-without-a-running-x-server

在 代码前面加上

import matplotlib

# Force matplotlib to not use any Xwindows backend.

matplotlib.use('Agg')

注意该代码一定要加载import matplotlib.pyplot as plt之前，ok即可正常保存

本地虚拟环境第二步无法运行，直接安装，其中将requirements_python3.txt中所有的==改为>=，安装完成所有的依赖库后，运行

python draw_convnet.py

无法保存图像,卡死，有如下三种解决方法

1、同样如上修改即可完成运行，

2、在弹出的画框中手动保存

3、删掉plt.show()即可避免该问题