mirror of
https://github.com/graphdeco-inria/gaussian-splatting
synced 2024-11-25 21:28:17 +00:00
fix bug for mask loading
This commit is contained in:
parent
ea68bdf29c
commit
933aa7d210
@ -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,
|
||||||
|
Loading…
Reference in New Issue
Block a user