utils.py 文件源码

python
阅读 22 收藏 0 点赞 0 评论 0

项目:FeatureMapInversion 作者: xzqjack 项目源码 文件源码
def SaveImage(img, args, epoch):
    """
    Postprocess Image and use total tv-norm to denoise postprocessed image

    1. postprocess Image
    2. use total tv-norm to denoise postprocessed image

    Parameters
    --------
    img: ndarray (1x3xMxN), optimized image

    Returns
    """
    out = PostprocessImage(img)
    out = denoise_tv_chambolle(out, weight=args.remove_noise, multichannel=True)
    if args.mod_type == "purposeful":
        save_name = os.path.join(args.output,"{}_{}_{}_{}_{}.jpg".\
        format(args.layer_name, args.mod_type, os.path.basename(args.content_image)[:-4],\
                 os.path.basename(args.style_image)[:-4], epoch))
    else:
        save_name = os.path.join(args.output,"{}_{}_{}_{}.jpg".\
        format(args.layer_name, args.mod_type, os.path.basename(args.content_image)[:-4], epoch))
    logging.info('save output to %s', save_name)
    io.imsave(save_name, out)
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号