embeddings

inference/cli_demo_camera_i2v_pcd.py

@@ -77,7 +77,7 @@ def maxpool_mask_tensor(mask_tensor):
         torch.Tensor: shape (12, 30, 45), pooled binary mask
     """
     T, H, W = mask_tensor.shape
-    assert T % 12 == 0, "T must be divisible by 12 (e.g., 48)"
+    # assert T % 12 == 0, "T must be divisible by 12 (e.g., 48)"
     # assert H % 30 == 0 and W % 45 == 0, "H and W must be divisible by 30 and 45"
     assert H % 8 == 0 and W % 8 == 0, "H and W must be divisible by 8 for spatial pooling"
 
@@ -111,7 +111,7 @@ def avgpool_mask_tensor(mask_tensor):
         torch.Tensor: shape (13, 30, 45), pooled binary mask with first frame zeroed
     """
     T, H, W = mask_tensor.shape
-    assert T % 12 == 0, "T must be divisible by 12 (e.g., 48)"
+    # assert T % 12 == 0, "T must be divisible by 12 (e.g., 48)"
     # assert H % 30 == 0 and W % 45 == 0, "H and W must be divisible by 30 and 45"
     assert H % 8 == 0 and W % 8 == 0, "H and W must be divisible by 8 for spatial pooling"