Update video_depth_anything/video_depth.py

video_depth_anything/video_depth.py

@@ -64,7 +64,7 @@ class VideoDepthAnything(nn.Module):
         depth = F.relu(depth)
         return depth.squeeze(1).unflatten(0, (B, T)) # return shape [B, T, H, W]
     
-    def infer_video_depth(self, frames, target_fps, input_size=518, device='cuda', progress=tqdm):
+    def infer_video_depth(self, frames, target_fps, input_size=518, device='cuda'):
         frame_height, frame_width = frames[0].shape[:2]
         ratio = max(frame_height, frame_width) / min(frame_height, frame_width)
         if ratio > 1.78:
@@ -95,7 +95,7 @@ class VideoDepthAnything(nn.Module):
         
         depth_list = []
         pre_input = None
-        for frame_id in progress(range(0, org_video_len, frame_step)):
+        for frame_id in tqdm(range(0, org_video_len, frame_step)):
             cur_list = []
             for i in range(INFER_LEN):
                 cur_list.append(torch.from_numpy(transform({'image': frame_list[frame_id+i].astype(np.float32) / 255.0})['image']).unsqueeze(0).unsqueeze(0))