add log

utils/dc_utils.py

@@ -16,6 +16,7 @@ from decord import VideoReader, cpu
 def read_video_frames(video_path, process_length, target_fps=-1, max_res=-1, dataset="open"):
 
     vid = VideoReader(video_path, ctx=cpu(0))
+    print("==> original video shape: ", (len(vid), *vid.get_batch([0]).shape[1:]))
     original_height, original_width = vid.get_batch([0]).shape[1:3]
     height = original_height
     width = original_width
@@ -32,6 +33,7 @@ def read_video_frames(video_path, process_length, target_fps=-1, max_res=-1, dat
     frames_idx = list(range(0, len(vid), stride))
     if process_length != -1 and process_length < len(frames_idx):
         frames_idx = frames_idx[:process_length]
+    print(f"==> final processing shape: {len(frames_idx), *vid.get_batch([0]).shape[1:]}")
     frames = vid.get_batch(frames_idx).asnumpy()
 
     return frames, fps

video_depth_anything/video_depth.py

@@ -71,6 +71,8 @@ class VideoDepthAnything(nn.Module):
             input_size = int(input_size * 1.78 / ratio)
             input_size = round(input_size / 14) * 14
 
+        print(f'==> infer lower bound input size: {input_size}')
+
         transform = Compose([
             Resize(
                 width=input_size,