From 81c5c4440fee07eed1d8255628ed35005b34d6a1 Mon Sep 17 00:00:00 2001 From: Allie Date: Wed, 19 Jun 2019 20:18:29 -0700 Subject: [PATCH] fix centerline nan issue (#6) --- demo_usage/cuboids_to_bboxes.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/demo_usage/cuboids_to_bboxes.py b/demo_usage/cuboids_to_bboxes.py index 085313d6..65d3f475 100644 --- a/demo_usage/cuboids_to_bboxes.py +++ b/demo_usage/cuboids_to_bboxes.py @@ -77,6 +77,9 @@ def plot_lane_centerlines_in_img( ground_heights = avm.get_ground_height_at_xy(centerline_city_fr, city_name) + valid_idx = np.isnan(ground_heights) + centerline_city_fr = centerline_city_fr[~valid_idx] + centerline_egovehicle_fr = city_to_egovehicle_se3.inverse().transform_point_cloud(centerline_city_fr) centerline_uv_cam = cam_SE3_egovehicle.transform_point_cloud(centerline_egovehicle_fr)