add imageio

app.py

@@ -10,7 +10,7 @@ import torch
 from decord import cpu, VideoReader
 from diffusers.training_utils import set_seed
 import torch.nn.functional as F
-# import imageio
+import imageio
 from kornia.filters import canny
 from kornia.morphology import dilation
 
@@ -185,8 +185,8 @@ def infer_geometry(
         output_disp_path.parent.mkdir(exist_ok=True)
             
         colored_disp = pmap_to_disp(point_maps, valid_masks)
-        # imageio.mimsave(
-        #     output_disp_path, (colored_disp*255).cpu().numpy().astype(np.uint8), fps=fps, macro_block_size=1)
+        imageio.mimsave(
+            output_disp_path, (colored_disp*255).cpu().numpy().astype(np.uint8), fps=fps, macro_block_size=1)
 
 
         # downsample for visualization

requirements.txt

@@ -15,3 +15,4 @@ moderngl
 piqp
 trimesh
 scipy
+imageio