当前位置: 首页>>代码示例>>Python>>正文


Python util.calculate_psnr方法代码示例

本文整理汇总了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 
开发者ID:sony,项目名称:nnabla-examples,代码行数:24,代码来源:train.py


注:本文中的utils.util.calculate_psnr方法示例由纯净天空整理自Github/MSDocs等开源代码及文档管理平台,相关代码片段筛选自各路编程大神贡献的开源项目,源码版权归原作者所有,传播和使用请参考对应项目的License;未经允许,请勿转载。