utils.py 文件源码

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)