slothfulxtx commited on
Commit
62df672
·
1 Parent(s): 925acf8

uemove imageio for testing

Browse files
Files changed (2) hide show
  1. app.py +3 -3
  2. 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
- output_disp_path, (colored_disp*255).cpu().numpy().astype(np.uint8), fps=fps, macro_block_size=1)
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