UniqAI/gaussian-splatting
1
0
Fork 0
mirror of https://github.com/graphdeco-inria/gaussian-splatting synced 2024-11-22 08:18:17 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix bug for mask loading

Browse Source
This commit is contained in:
bopan3 2023-09-07 21:19:47 +08:00
parent ea68bdf29c
commit 933aa7d210
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils/camera_utils.py
View File

@ -43,7 +43,7 @@ def loadCam(args, id, cam_info, resolution_scale):
gt_image = resized_image_rgb[:3, ...] gt_image = resized_image_rgb[:3, ...]
loaded_mask = None loaded_mask = None
if resized_image_rgb.shape[1] == 4: if resized_image_rgb.shape[0] == 4:
loaded_mask = resized_image_rgb[3:4, ...] loaded_mask = resized_image_rgb[3:4, ...]
return Camera(colmap_id=cam_info.uid, R=cam_info.R, T=cam_info.T, return Camera(colmap_id=cam_info.uid, R=cam_info.R, T=cam_info.T,