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

参考博文: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)