本文整理汇总了Python中utils.util.calculate_psnr方法的典型用法代码示例。如果您正苦于以下问题：Python util.calculate_psnr方法的具体用法？Python util.calculate_psnr怎么用？Python util.calculate_psnr使用的例子？那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类utils.util的用法示例。
在下文中一共展示了util.calculate_psnr方法的1个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: val_save
# 需要导入模块: from utils import util [as 别名]
# 或者: from utils.util import calculate_psnr [as 别名]
def val_save(val_gt, val_lq, val_lq_path, idx, epoch, avg_psnr):
conf = get_config()
sr_img = rrdb_net(val_lq, 64, 23)
real_image = array_to_image(val_gt.data)
sr_image = array_to_image(sr_img.data)
img_name = os.path.splitext(
os.path.basename(val_lq_path[idx]))[0]
img_dir = os.path.join(
conf.val.save_results + "/results", img_name)
if not os.path.exists(img_dir):
os.makedirs(img_dir)
save_img_path = os.path.join(
img_dir, '{:s}_{:d}.png'.format(img_name, epoch))
cv2.imwrite(save_img_path, sr_image)
crop_size = conf.train.scale
cropped_sr_image = sr_image[crop_size:-
crop_size, crop_size:-crop_size, :]
cropped_real_image = real_image[crop_size:-
crop_size, crop_size:-crop_size, :]
avg_psnr += calculate_psnr(cropped_sr_image, cropped_real_image)
print("validating", img_name)
return avg_psnr