def save_img(img, mse_diff, nrmse_diff, ssim_diff, psnr_diff):
p = os.path.join(
save_dir, "wgan-%.4f-%.4f-%.4f-%.4f.png" %
(ssim_diff, mse_diff, nrmse_diff, psnr_diff))
save_image(img, img_path=p)
print("wgan generated ssim: %.4f images saved at %s" %
(ssim_diff, p))
def save_single_img(img, count, bt):
p = os.path.join(save_dir, "cgan_single_%d_%d.png" % (count, bt))
save_image(img, img_path=p)
print("cgan single sample id: %d _ %d saved" % (count, bt))
def save_img(img, mse_diff, nrmse_diff, ssim_diff, psnr_diff):
p = os.path.join(save_dir, "cgan_patch%.4f-%.4f-%.4f-%.4f.png" % (ssim_diff, mse_diff, nrmse_diff,
psnr_diff))
save_image(img, img_path=p)