1.打开txt文件
2.写入数据
3.关闭文件夹
eval_summary = open(os.path.join(args.output_dir, "metrics.txt"),"w")
for ...... :
.........
print(
'Name:{} | PSNR:{:4f} | SSIM:{:4f} '.format(idx, cur_psnr, cur_ssim)
)
eval_summary.write('Name:{} | PSNR:{:4f} | SSIM:{:4f} \n'.format(idx, cur_psnr, cur_ssim))
eval_summary.close()
效果如图: