神经网络结构画图

参考代码：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()即可避免该问题