pytorch中怎么用plt显示tensor？
- 后端
- 猿友
- 0
在pytorch中图片的张量结构与plt可以显示的图片格式要求是不一样的，所以plt是不能直接显示tensor格式的图片的，那么pytorch怎么用plt显示tensor图片呢？这就需要涉及到数据转换了，基本思路就是将tensor转换为numpy类型的数据结构，而numpy类型的格式刚好可以被plt支持。接下来就来看具体怎么操作吧！
问题
图像的张量结构为（C，H，W），而plt可以显示的图片格式要求（H，W，C），C为颜色通道数，可以没有。
所以问题就是将Tensor（C，H，W）=> numpy（H，W，C）
解决办法
def transimg(img):
img = img / 2 + 0.5 # unnormalize
npimg = img.numpy()
npimg1 = np.transpose(npimg,(1,2,0)) # C*H*W => H*W*C
return npimg1
以上就是pytorch怎么用plt显示tensor的方法介绍了，希望能给大家一个参考，也希望大家多多支持W3Cschool。
免责申明：本站发布的内容（图片、视频和文字）以转载和分享为主，文章观点不代表本站立场，如涉及侵权请联系站长邮箱：xbc-online@qq.com进行反馈，一经查实，将立刻删除涉嫌侵权内容。
