0
点赞
收藏
分享

微信扫一扫

[PyTorch] 可视化tensor图像

你的益达233 2022-01-23 阅读 106

PyTorch提供了直接对tensor可视化的接口函数:

plt.imshow(transforms.ToPILImage()(image), interpolation="bicubic")
transforms.ToPILImage()(image).show() # Alternatively

参考:How do I display a single image in PyTorch?

举报

相关推荐

0 条评论