Update utility.py

yulunzhang · Sep 17, 2019 · c4d133d · c4d133d
1 parent b90f676
commit c4d133d
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/RCAN_TestCode/code/utility.py b/RCAN_TestCode/code/utility.py
@@ -9,7 +9,8 @@
 import matplotlib.pyplot as plt
 
 import numpy as np
-import scipy.misc as misc
+# import scipy.misc as misc
+import imageio
 
 import torch
 import torch.optim as optim
@@ -124,7 +125,8 @@ def save_results(self, filename, save_list, scale):
         for v, p in zip(save_list, postfix):
             normalized = v[0].data.mul(255 / self.args.rgb_range)
             ndarr = normalized.byte().permute(1, 2, 0).cpu().numpy()
-            misc.imsave('{}{}.png'.format(filename, p), ndarr)
+            #misc.imsave('{}{}.png'.format(filename, p), ndarr)
+            imageio.imsave('{}{}.png'.format(filename, p), ndarr)
 
     def save_results_nopostfix(self, filename, save_list, scale):
         #print(filename)
@@ -138,7 +140,8 @@ def save_results_nopostfix(self, filename, save_list, scale):
         for v, p in zip(save_list, postfix):
             normalized = v[0].data.mul(255 / self.args.rgb_range)
             ndarr = normalized.byte().permute(1, 2, 0).cpu().numpy()
-            misc.imsave('{}.png'.format(filename), ndarr)
+            #misc.imsave('{}.png'.format(filename), ndarr)
+            imageio.imsave('{}.png'.format(filename), ndarr)
 
 
 def quantize(img, rgb_range):