Update test.py

Author: dmhdmhdmh

test.py

@@ -117,7 +117,7 @@ def eval_psnr(loader, model, data_norm=None, eval_type=None, eval_bsize=None, wi
             pred = pred[..., :batch['gt'].shape[-2], :batch['gt'].shape[-1]]
             predsave = img_as_ubyte(np.clip(pred.cpu().numpy()[0], 0, 1)).transpose((1, 2, 0))
         name = batch[k]
-#       io.imsave('/root/autodl-tmp/tcgatextureattensave/X2/' + name[0], predsave)
+        io.imsave('/root/autodl-tmp/tcgatextureattensave/X2/' + name[0], predsave)
         res = metric_fn(pred, batch['gt'])
         val_res.add(res.item(), inp.shape[0])
 
@@ -159,4 +159,4 @@ def eval_psnr(loader, model, data_norm=None, eval_type=None, eval_bsize=None, wi
                     scale_max=int(args.scale_max),
                     fast=args.fast,
                     verbose=True)
-    print('result: {:.4f}'.format(res))
+    #print('result: {:.4f}'.format(res))