fix pgd config

open-mmlab · Dec 28, 2023 · d714153 · d714153
1 parent 80a7a77
commit d714153
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 8 deletions.
diff --git a/configs/_base_/datasets/waymoD3-fov-mono3d-3class.py b/configs/_base_/datasets/waymoD3-fov-mono3d-3class.py
@@ -173,12 +173,10 @@
 
 val_evaluator = dict(
     type='WaymoMetric',
-    ann_file='./data/waymo/kitti_format/waymo_infos_val.pkl',
     waymo_bin_file='./data/waymo/waymo_format/fov_gt.bin',
-    pklfile_prefix='./pgd_fov_pred',
     metric='LET_mAP',
-    convert_kitti_format=False,
-    backend_args=backend_args)
+    load_type='fov_image_based',
+    result_prefix='./pgd_fov_pred')
 test_evaluator = val_evaluator
 
 vis_backends = [dict(type='LocalVisBackend')]

diff --git a/configs/_base_/datasets/waymoD3-mv-mono3d-3class.py b/configs/_base_/datasets/waymoD3-mv-mono3d-3class.py
@@ -172,13 +172,18 @@
 
 val_evaluator = dict(
     type='WaymoMetric',
-    ann_file='./data/waymo/kitti_format/waymo_infos_val.pkl',
     waymo_bin_file='./data/waymo/waymo_format/cam_gt.bin',
-    pklfile_prefix='./pgd_mv',
     metric='LET_mAP',
-    convert_kitti_format=False,
     load_type='mv_image_based',
-    backend_args=backend_args)
+    result_prefix='./pgd_mv_pred',
+    nms_cfg=dict(
+        use_rotate_nms=True,
+        nms_across_levels=False,
+        nms_pre=500,
+        nms_thr=0.05,
+        score_thr=0.001,
+        min_bbox_size=0,
+        max_per_frame=100))
 test_evaluator = val_evaluator
 
 vis_backends = [dict(type='LocalVisBackend')]