PyCharm中使用display.Image无法显示Caffe2网络结构图像求助

Hey，这个问题我在PyCharm里踩过坑，给你几个靠谱的解决办法：

1. 开启PyCharm的科学绘图工具窗口 #

PyCharm默认可能没启用内联图像渲染，按以下步骤设置就能搞定：

• 打开PyCharm的File -> Settings（Windows/Linux）或者PyCharm -> Settings（Mac）
• 导航到Tools -> Python Scientific选项
• 勾选Show plots in toolwindow，点击OK保存设置
• 重启当前的IPython内核，再重新运行代码，display.Image应该就能正常显示图像了

2. 手动保存图像并打开 #

如果调整设置后还是不行，直接把生成的图像存到本地再查看更稳妥。修改代码最后几行：

graph = net_drawer.GetPydotGraph(net, rankdir="LR")
# 将图像保存为本地PNG文件
graph.write_png("my_network_structure.png")

# 调用系统默认图片查看器打开
from PIL import Image
img = Image.open("my_network_structure.png")
img.show()

执行后，你会在项目根目录找到my_network_structure.png，同时系统会自动弹出窗口显示这个网络结构图。

3. 检查依赖是否完整 #

有时候图像无法生成是因为缺少底层依赖，确认以下几点：

• 确保安装了Python的pydotplus和graphviz包：

  pip install pydotplus graphviz
  

• 安装系统级的Graphviz库：
  • Ubuntu/Debian：sudo apt-get install graphviz
  • Mac：brew install graphviz
  • Windows：去Graphviz官方网站下载安装包，安装后记得把bin目录添加到系统环境变量PATH中

你看到的<IPython.core.display.Image object>是因为PyCharm的IPython环境没有正确解析display模块的输出，只是打印了对象的字符串表示，而不是渲染成图像。上面的方法都能解决这个问题。

内容的提问来源于stack exchange，提问作者Oliver