如何利用Python保存生成的图片

最新推荐文章于 2023-04-20 19:42:56 发布

波尔德

最新推荐文章于 2023-04-20 19:42:56 发布

阅读量973

点赞数

分类专栏：笔记学习生活科研日常文章标签： python numpy 开发语言

本文链接：https://blog.csdn.net/weixin_44025103/article/details/128328322

版权

学习生活同时被 3 个专栏收录

129 篇文章

订阅专栏

笔记

112 篇文章

订阅专栏

科研日常

68 篇文章

订阅专栏

该博文详细介绍了如何在PyTorch中利用torchvision.utils.save_image函数将tensor类型图片保存到本地。首先检查保存路径是否存在，然后依次保存原始图片、 ground truth 和预测结果，每个图片都对应一个唯一的文件名，包括当前epoch编号。

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

参考博文:https://blog.csdn.net/weixin_38645650/article/details/121704001
使用的API:

== torchvision.utils.save_image（tensor类型的图片，保存的图片路径）==
具体代码如下:

     # 如果文件夹不存在，就建一个文件夹
        if not os.path.isdir(valid_save_path ):
            os.makedirs(valid_save_path )
        # 确定图片的保存路径以及图片名称
        for i in range():
            filepath_img=valid_save_path+"/"+ str(now_epoch_num)+"_img.png"  # filename:'./results/hiformer-s/valid/valid1_img.png'
            # torchvision 只能保存tensor类型的图片。
            torchvision.utils.save_image(original_img,filepath_img)
            filepath_gt = valid_save_path + "/" + str(now_epoch_num) + "_gt.png"
            torchvision.utils.save_image(original_label,filepath_gt)
            filepath_prd = valid_save_path+'/'+str(now_epoch_num)+"_prd.png"
            prediction = torch.from_numpy(prediction)
            torchvision.utils.save_image(prediction,filepath_prd)