fix: minor waymo eval class name typo (#1136)

8f765b66 · Jungwan Woo · GitHub · b345b08c · 8f765b66
Unverified Commit 8f765b66 authored Oct 06, 2022 by Jungwan Woo Committed by GitHub Oct 06, 2022
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

pcdet/datasets/waymo/waymo_eval.py pcdet/datasets/waymo/waymo_eval.py +1 -1

No files found.
--- a/pcdet/datasets/waymo/waymo_eval.py
+++ b/pcdet/datasets/waymo/waymo_eval.py
@@ -21,7 +21,7 @@ def limit_period(val, offset=0.5, period=np.pi):


 class OpenPCDetWaymoDetectionMetricsEstimator(tf.test.TestCase):
-    WAYMO_CLASSES = ['unknown', 'Vehicle', 'Pedestrian', 'Truck', 'Cyclist']
+    WAYMO_CLASSES = ['unknown', 'Vehicle', 'Pedestrian', 'Sign', 'Cyclist']

    def generate_waymo_type_results(self, infos, class_names, is_gt=False, fake_gt_infos=True):
        def boxes3d_kitti_fakelidar_to_lidar(boxes3d_lidar):