本文整理匯總了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