pytorch训练过程中存图片的一种方式

FrenchOldDriver

于 2023-02-28 22:08:44 发布

阅读量326

点赞数 1

CC 4.0 BY-SA版权

分类专栏：图像处理深度学习文章标签： pytorch python 计算机视觉

本文链接：https://blog.csdn.net/OldDriver1995/article/details/129271179

图像处理同时被 2 个专栏收录

15 篇文章

订阅专栏

深度学习

11 篇文章

订阅专栏

这段代码展示了如何利用einops的rearrange函数将数据从chw格式转换到hwc格式，然后用PIL的Image模块创建并保存图像。转换过程适用于numpy数组，并提及opencv和plt作为其他可能的图像处理选项。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

个人随笔记录，方便使用

from einops import rearrange
from PIL import Image
x_sample = 255. * rearrange(x_sample.cpu().numpy(), 'c h w -> h w c')
img = Image.fromarray(x_sample.astype(np.uint8))
img.save("test.png")

用其他的opencv或者plt应该也是可以的