tensor图像的快速保存

暗巷提灯

于 2024-08-25 13:36:07 发布

阅读量300

点赞数 2

文章标签： tensor 深度学习图像保存 cv

本文链接：https://blog.csdn.net/m0_75015083/article/details/141528452

版权

"""
假设image_A是一张[1,3,256,256]的tensor数据，一般方法就是先转化为numpy，在通过numpy中的
相关方法进行保存
"""
image_A=torch.squeeze(image_A, dim=0)
#permute是用来调换维度顺序的函数
image_A = image_A.permute(1, 2, 0).cpu().numpy() * 255  # 转换为 [256, 256, 3] 形状
image_A = Image.fromarray(image_A.astype(np.uint8))
# 保存图像
image_A.save(os.path.join(path_name, 'image_A.jpg'))

通过调整维度便可以快速使用