bugfixed: set empty gt_boxes_lidar to (0, 9) instead of (0, 7)

thri5ha · Aug 18, 2022 · 6a1f253 · 6a1f253
1 parent 4996eb4
commit 6a1f253
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pcdet/datasets/waymo/waymo_utils.py b/pcdet/datasets/waymo/waymo_utils.py
@@ -65,7 +65,7 @@ def generate_labels(frame, pose):
             axis=1
         )
     else:
-        gt_boxes_lidar = np.zeros((0, 7))
+        gt_boxes_lidar = np.zeros((0, 9))
     annotations['gt_boxes_lidar'] = gt_boxes_lidar
     return annotations