Python 有没有一个通用的图片展示包

[KKKg]

这几天写代码，一到展示图片就头疼。有的读进来是 C H W ，有的是 H W C ，tensor 要送 numpy 和 cpu ，如果在 0 到 1 还要手动乘 255 。很多时候我只是想随手看一眼图片而已，要求不高，有没有造好的轮子可以通用地展示图片，不去操心 shape 之类的事情？

[julyclyde]

pillow 可以吗

[joApioVVx4M4X6Rf]

搞到 HTML 里面不行吗

[KKKg]

@julyclyde Pillow 只有读图片文件的功能吧，貌似没有面向 torch 和 np 的函数

[noahliaszn]

opencv?

[Belmode]

为什么...这么多字连在一起，我看不懂了......🤣

[clino]

面向 torch 和 np 的图片展示，算"通用"吗？明明是给这俩量身定制的吧

[winglight2016]

你的需求真奇怪，只是想随手看一眼图片，为什么不双击用系统默认程序打开看？

[pelloz]

说实话，我看不懂你在说啥...

[hsfzxjy]

@pelloz @Belmode 就是说各个库里图片的表示都不一样，有的 channel 在前有的在后，有的 0-255 有的 0-1 。楼主想找一个能同时处理并展示这些图片的方法

[hsfzxjy]

@winglight2016 可能是程序算出来的中间结果，落盘再看太麻烦

[Muniesa]

vscode 有个扩展 https://github.com/elazarcoh/simply-view-image-for-python-debugging

[haha512]

写个函数兼容下。shape[0]如果是 1 或者 3 ，并且 shape[2]不是 1 或 3 ，那么就是 c 在前。否则就是 c 在后，当然可能存在 w 是 1 或 3 像素的图，但应该小概率吧

[Jimmyisme]

@winglight2016 有时候可能图片是保存的 npy 或者 pkl

[shxxy]

呃，matplotlib 我记得不就可以吗

[KKKg]

@Muniesa 多谢！我看看 pycharm 有没有类似插件

[KKKg]

@haha512 是的，，下午写了一个类似的函数

[FlashEcho]

让 chatgpt 帮你写，告诉他是 ndarrary 还是 tensor 就行了，channel 在哪个维度，面向 ai 编程
这种技术难度不高的需求，chatgpt 正确率很高的