Fix crop obb masking in pointcloud generation (#3433)

This commit is contained in:
Nicholas Pfaff authored and GitHub committed 2024-09-16 16:44:31 -04:00
1 parent 106d427e80
commit 27b8e14798
1 file changed
+5 -5
+5 -5
View File
@@ -165,11 +165,11 @@ def generate_point_cloud(
if crop_obb is not None:
mask = crop_obb.within(point)
point = point[mask]
rgb = rgb[mask]
view_direction = view_direction[mask]
if normal is not None:
normal = normal[mask]
point = point[mask]
rgb = rgb[mask]
view_direction = view_direction[mask]
if normal is not None:
normal = normal[mask]
points.append(point)
rgbs.append(rgb)