fix bug for mask loading

This commit is contained in:
bopan3 2023-09-07 21:19:47 +08:00
parent ea68bdf29c
commit 933aa7d210

View File

@ -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,