VicFonch
commited on
README.md Update
Browse files
README.md
CHANGED
@@ -63,9 +63,11 @@ tau = 0.5
|
|
63 |
img1 = model.reverse_process([img0, img2], tau)
|
64 |
|
65 |
plt.figure(figsize=(10, 5))
|
66 |
-
plt.subplot(1,
|
67 |
plt.imshow(denorm(img0, mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5]).squeeze().permute(1, 2, 0).cpu().numpy())
|
68 |
-
plt.subplot(1,
|
69 |
-
plt.imshow(denorm(
|
|
|
|
|
70 |
plt.show()
|
71 |
```
|
|
|
63 |
img1 = model.reverse_process([img0, img2], tau)
|
64 |
|
65 |
plt.figure(figsize=(10, 5))
|
66 |
+
plt.subplot(1, 3, 1)
|
67 |
plt.imshow(denorm(img0, mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5]).squeeze().permute(1, 2, 0).cpu().numpy())
|
68 |
+
plt.subplot(1, 3, 2)
|
69 |
+
plt.imshow(denorm(img1, mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5]).squeeze().permute(1, 2, 0).cpu().numpy())
|
70 |
+
plt.subplot(1, 3, 3)
|
71 |
+
plt.imshow(denorm(img2, mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5]).squeeze().permute(1, 2, 0).cpu().numpy())
|
72 |
plt.show()
|
73 |
```
|