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)
评论列表
文章目录