hail75 commited on
Commit
45b47cc
·
1 Parent(s): 828ec7e
Files changed (1) hide show
  1. models/SRFlow/srflow.py +3 -3
models/SRFlow/srflow.py CHANGED
@@ -69,8 +69,8 @@ if __name__ == '__main__':
69
 
70
  sr = return_SRFlow_result_from_tensor(lr_tensor)
71
  print(sr.shape)
72
-
 
73
  plt.imshow(np.transpose(sr[0].cpu().detach().numpy(), (1, 2, 0)))
74
- plt.axis('off')
75
- plt.title('Super-Resolved Image')
76
  plt.show()
 
69
 
70
  sr = return_SRFlow_result_from_tensor(lr_tensor)
71
  print(sr.shape)
72
+
73
+ # Show SR image of the first one in the batch
74
  plt.imshow(np.transpose(sr[0].cpu().detach().numpy(), (1, 2, 0)))
75
+ # plt.axis('off')
 
76
  plt.show()