Spaces:
Running
Running
small fix
Browse files- 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()
|