Spaces:
Running
on
Zero
Running
on
Zero
slothfulxtx
commited on
Commit
·
62df672
1
Parent(s):
925acf8
uemove imageio for testing
Browse files- app.py +3 -3
- requirements.txt +1 -1
app.py
CHANGED
@@ -10,7 +10,7 @@ import torch
|
|
10 |
from decord import cpu, VideoReader
|
11 |
from diffusers.training_utils import set_seed
|
12 |
import torch.nn.functional as F
|
13 |
-
import imageio
|
14 |
from kornia.filters import canny
|
15 |
from kornia.morphology import dilation
|
16 |
|
@@ -183,8 +183,8 @@ def infer_geometry(
|
|
183 |
output_disp_path.parent.mkdir(exist_ok=True)
|
184 |
|
185 |
colored_disp = pmap_to_disp(point_maps, valid_masks)
|
186 |
-
imageio.mimsave(
|
187 |
-
|
188 |
|
189 |
|
190 |
# downsample for visualization
|
|
|
10 |
from decord import cpu, VideoReader
|
11 |
from diffusers.training_utils import set_seed
|
12 |
import torch.nn.functional as F
|
13 |
+
# import imageio
|
14 |
from kornia.filters import canny
|
15 |
from kornia.morphology import dilation
|
16 |
|
|
|
183 |
output_disp_path.parent.mkdir(exist_ok=True)
|
184 |
|
185 |
colored_disp = pmap_to_disp(point_maps, valid_masks)
|
186 |
+
# imageio.mimsave(
|
187 |
+
# output_disp_path, (colored_disp*255).cpu().numpy().astype(np.uint8), fps=fps, macro_block_size=1)
|
188 |
|
189 |
|
190 |
# downsample for visualization
|
requirements.txt
CHANGED
@@ -14,4 +14,4 @@ opencv-python==4.10.0.84
|
|
14 |
h5py==3.12.1
|
15 |
moderngl==5.12.0
|
16 |
piqp==0.4.2
|
17 |
-
imageio==2.36.0
|
|
|
14 |
h5py==3.12.1
|
15 |
moderngl==5.12.0
|
16 |
piqp==0.4.2
|
17 |
+
# imageio==2.36.0
|