Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
3c9122e
1
Parent(s):
2839355
change image grid order
Browse files
app.py
CHANGED
@@ -160,7 +160,8 @@ def generate_mvs(input_image, sample_steps, sample_seed):
|
|
160 |
|
161 |
show_image = np.asarray(z123_image, dtype=np.uint8)
|
162 |
show_image = torch.from_numpy(show_image) # (960, 640, 3)
|
163 |
-
show_image = rearrange(show_image, '(n h) (m w) c -> (m h
|
|
|
164 |
show_image = Image.fromarray(show_image.numpy())
|
165 |
|
166 |
return z123_image, show_image
|
|
|
160 |
|
161 |
show_image = np.asarray(z123_image, dtype=np.uint8)
|
162 |
show_image = torch.from_numpy(show_image) # (960, 640, 3)
|
163 |
+
show_image = rearrange(show_image, '(n h) (m w) c -> (n m) h w c', n=3, m=2)
|
164 |
+
show_image = rearrange(show_image, '(n m) h w c -> (n h) (m w) c', n=2, m=3)
|
165 |
show_image = Image.fromarray(show_image.numpy())
|
166 |
|
167 |
return z123_image, show_image
|