From 27b8e14798a49ab763d978bcb732916896618fc9 Mon Sep 17 00:00:00 2001 From: Nicholas Pfaff <53228351+nepfaff@users.noreply.github.com> Date: Mon, 16 Sep 2024 16:44:31 -0400 Subject: [PATCH] Fix crop obb masking in pointcloud generation (#3433) --- nerfstudio/exporter/exporter_utils.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/nerfstudio/exporter/exporter_utils.py b/nerfstudio/exporter/exporter_utils.py index b87078bc9..9435a3b91 100644 --- a/nerfstudio/exporter/exporter_utils.py +++ b/nerfstudio/exporter/exporter_utils.py @@ -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)