From 933aa7d2100284841bb6c2be520c412b4e64335a Mon Sep 17 00:00:00 2001
From: bopan3 <1450662430@qq.com>
Date: Thu, 7 Sep 2023 21:19:47 +0800
Subject: [PATCH] fix bug for mask loading

---
 utils/camera_utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/utils/camera_utils.py b/utils/camera_utils.py
index 1a54d0a..bdc2e9b 100644
--- a/utils/camera_utils.py
+++ b/utils/camera_utils.py
@@ -43,7 +43,7 @@ def loadCam(args, id, cam_info, resolution_scale):
     gt_image = resized_image_rgb[:3, ...]
     loaded_mask = None
 
-    if resized_image_rgb.shape[1] == 4:
+    if resized_image_rgb.shape[0] == 4:
         loaded_mask = resized_image_rgb[3:4, ...]
 
     return Camera(colmap_id=cam_info.uid, R=cam_info.R, T=cam_info.T,