VicFonch commited on
Commit
f82b414
·
unverified ·
1 Parent(s): 6398d0b

README.md Update

Browse files
Files changed (1) hide show
  1. README.md +5 -3
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, 2, 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, 2, 2)
69
- plt.imshow(denorm(It, mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5]).squeeze().permute(1, 2, 0).cpu().numpy())
 
 
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
  ```