modify to exclude instances that only have 2D annotations

Signed-off-by: Shunsuke Miura <[email protected]>
tier4 · Nov 23, 2023 · 8e017c0 · 8e017c0
1 parent 3c39367
commit 8e017c0
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/perception_dataset/utils/calculate_num_points.py b/perception_dataset/utils/calculate_num_points.py
@@ -59,6 +59,9 @@ def calculate_num_points(
 
     # connect next/prev tokens
     for instance in nusc.instance:
+        if instance["nbr_annotations"] == 0:
+            logger.warning(f"instance:{instance['token']} has no 3D annotation")
+            continue
         try:
             prev_sample_data: str = annotation_table._token_to_record[
                 instance["first_annotation_token"]