s

dkt/pipelines/pipeline.py

@@ -990,7 +990,18 @@ class DKTPipeline:
                 moge_mask = output['mask'].cpu().numpy()
                 moge_depth = output['depth'].cpu().numpy()
 
+                # Debug: check moge_mask
+                logger.info(f'moge_mask dtype: {moge_mask.dtype}, shape: {moge_mask.shape}, valid_count: {moge_mask.sum()}, total: {moge_mask.size}')
+                print(f'moge_mask dtype: {moge_mask.dtype}, shape: {moge_mask.shape}, valid_count: {moge_mask.sum()}, total: {moge_mask.size}')
+                
+                # Ensure moge_mask is boolean type
+                moge_mask = moge_mask.astype(bool)
+
                 metric_depth, scale, shift = transfer_pred_disp2depth(predicted_depth, moge_depth, moge_mask, return_scale_shift=True)
+                
+                # Debug: check metric_depth
+                logger.info(f'metric_depth shape: {metric_depth.shape}, min: {metric_depth.min():.4f}, max: {metric_depth.max():.4f}')
+                print(f'metric_depth shape: {metric_depth.shape}, min: {metric_depth.min():.4f}, max: {metric_depth.max():.4f}')
 
                 moge_intrinsics[0, 0] *= resize_W 
                 moge_intrinsics[1, 1] *= resize_H