神经网络结构画图

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

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容