update configs

Thinklab-SJTU · Aug 15, 2024 · a630480 · a630480
1 parent a95fdcb
commit a630480
Show file tree

Hide file tree

Showing 7 changed files with 86 additions and 14 deletions.
diff --git a/adzoo/bevformer/configs/bevformer/bevformer_base_b2d.py b/adzoo/bevformer/configs/bevformer/bevformer_base_b2d.py
@@ -11,17 +11,14 @@
 point_cloud_range = [-51.2, -51.2, -5.0, 51.2, 51.2, 3.0]
 voxel_size = [0.2, 0.2, 8]
 
-
-
-
-
 NameMapping = {
     #=================vehicle=================
     # bicycle
     'vehicle.bh.crossbike': 'bicycle',
     "vehicle.diamondback.century": 'bicycle',
     "vehicle.gazelle.omafiets": 'bicycle',
     # car
+    "vehicle.audi.etron": 'car',
     "vehicle.chevrolet.impala": 'car',
     "vehicle.dodge.charger_2020": 'car',
     "vehicle.dodge.charger_police": 'car',
@@ -77,19 +74,30 @@
 
     #=================pedestrian==============
     "walker.pedestrian.0001": 'pedestrian',
+    "walker.pedestrian.0003": 'pedestrian',
     "walker.pedestrian.0004": 'pedestrian',
     "walker.pedestrian.0005": 'pedestrian',
     "walker.pedestrian.0007": 'pedestrian',
+    "walker.pedestrian.0010": 'pedestrian',
     "walker.pedestrian.0013": 'pedestrian',
     "walker.pedestrian.0014": 'pedestrian',
+    "walker.pedestrian.0015": 'pedestrian',
+    "walker.pedestrian.0016": 'pedestrian',
     "walker.pedestrian.0017": 'pedestrian',
     "walker.pedestrian.0018": 'pedestrian',
     "walker.pedestrian.0019": 'pedestrian',
     "walker.pedestrian.0020": 'pedestrian',
+    "walker.pedestrian.0021": 'pedestrian',
     "walker.pedestrian.0022": 'pedestrian',
     "walker.pedestrian.0025": 'pedestrian',
+    "walker.pedestrian.0027": 'pedestrian',
+    "walker.pedestrian.0030": 'pedestrian',
+    "walker.pedestrian.0031": 'pedestrian',
+    "walker.pedestrian.0032": 'pedestrian',
+    "walker.pedestrian.0034": 'pedestrian',
     "walker.pedestrian.0035": 'pedestrian',
     "walker.pedestrian.0041": 'pedestrian',
+    "walker.pedestrian.0042": 'pedestrian',
     "walker.pedestrian.0046": 'pedestrian',
     "walker.pedestrian.0047": 'pedestrian',
 
@@ -98,9 +106,6 @@
     "static.prop.dirtdebris02": 'others',
 }
 
-
-
-
 eval_cfg = {
             "dist_ths": [0.5, 1.0, 2.0, 4.0],
             "dist_th_tp": 2.0,

diff --git a/adzoo/bevformer/configs/bevformer/bevformer_tiny_b2d.py b/adzoo/bevformer/configs/bevformer/bevformer_tiny_b2d.py
@@ -11,17 +11,14 @@
 point_cloud_range = [-51.2, -51.2, -5.0, 51.2, 51.2, 3.0]
 voxel_size = [0.2, 0.2, 8]
 
-
-
-
-
 NameMapping = {
     #=================vehicle=================
     # bicycle
     'vehicle.bh.crossbike': 'bicycle',
     "vehicle.diamondback.century": 'bicycle',
     "vehicle.gazelle.omafiets": 'bicycle',
     # car
+    "vehicle.audi.etron": 'car',
     "vehicle.chevrolet.impala": 'car',
     "vehicle.dodge.charger_2020": 'car',
     "vehicle.dodge.charger_police": 'car',
@@ -77,19 +74,30 @@
 
     #=================pedestrian==============
     "walker.pedestrian.0001": 'pedestrian',
+    "walker.pedestrian.0003": 'pedestrian',
     "walker.pedestrian.0004": 'pedestrian',
     "walker.pedestrian.0005": 'pedestrian',
     "walker.pedestrian.0007": 'pedestrian',
+    "walker.pedestrian.0010": 'pedestrian',
     "walker.pedestrian.0013": 'pedestrian',
     "walker.pedestrian.0014": 'pedestrian',
+    "walker.pedestrian.0015": 'pedestrian',
+    "walker.pedestrian.0016": 'pedestrian',
     "walker.pedestrian.0017": 'pedestrian',
     "walker.pedestrian.0018": 'pedestrian',
     "walker.pedestrian.0019": 'pedestrian',
     "walker.pedestrian.0020": 'pedestrian',
+    "walker.pedestrian.0021": 'pedestrian',
     "walker.pedestrian.0022": 'pedestrian',
     "walker.pedestrian.0025": 'pedestrian',
+    "walker.pedestrian.0027": 'pedestrian',
+    "walker.pedestrian.0030": 'pedestrian',
+    "walker.pedestrian.0031": 'pedestrian',
+    "walker.pedestrian.0032": 'pedestrian',
+    "walker.pedestrian.0034": 'pedestrian',
     "walker.pedestrian.0035": 'pedestrian',
     "walker.pedestrian.0041": 'pedestrian',
+    "walker.pedestrian.0042": 'pedestrian',
     "walker.pedestrian.0046": 'pedestrian',
     "walker.pedestrian.0047": 'pedestrian',
 
@@ -98,9 +106,6 @@
     "static.prop.dirtdebris02": 'others',
 }
 
-
-
-
 eval_cfg = {
             "dist_ths": [0.5, 1.0, 2.0, 4.0],
             "dist_th_tp": 2.0,

diff --git a/adzoo/uniad/configs/stage1_track_map/base_track_map_b2d.py b/adzoo/uniad/configs/stage1_track_map/base_track_map_b2d.py
@@ -27,6 +27,7 @@
     "vehicle.diamondback.century": 'bicycle',
     "vehicle.gazelle.omafiets": 'bicycle',
     # car
+    "vehicle.audi.etron": 'car',
     "vehicle.chevrolet.impala": 'car',
     "vehicle.dodge.charger_2020": 'car',
     "vehicle.dodge.charger_police": 'car',
@@ -82,19 +83,30 @@
 
     #=================pedestrian==============
     "walker.pedestrian.0001": 'pedestrian',
+    "walker.pedestrian.0003": 'pedestrian',
     "walker.pedestrian.0004": 'pedestrian',
     "walker.pedestrian.0005": 'pedestrian',
     "walker.pedestrian.0007": 'pedestrian',
+    "walker.pedestrian.0010": 'pedestrian',
     "walker.pedestrian.0013": 'pedestrian',
     "walker.pedestrian.0014": 'pedestrian',
+    "walker.pedestrian.0015": 'pedestrian',
+    "walker.pedestrian.0016": 'pedestrian',
     "walker.pedestrian.0017": 'pedestrian',
     "walker.pedestrian.0018": 'pedestrian',
     "walker.pedestrian.0019": 'pedestrian',
     "walker.pedestrian.0020": 'pedestrian',
+    "walker.pedestrian.0021": 'pedestrian',
     "walker.pedestrian.0022": 'pedestrian',
     "walker.pedestrian.0025": 'pedestrian',
+    "walker.pedestrian.0027": 'pedestrian',
+    "walker.pedestrian.0030": 'pedestrian',
+    "walker.pedestrian.0031": 'pedestrian',
+    "walker.pedestrian.0032": 'pedestrian',
+    "walker.pedestrian.0034": 'pedestrian',
     "walker.pedestrian.0035": 'pedestrian',
     "walker.pedestrian.0041": 'pedestrian',
+    "walker.pedestrian.0042": 'pedestrian',
     "walker.pedestrian.0046": 'pedestrian',
     "walker.pedestrian.0047": 'pedestrian',
 

diff --git a/adzoo/uniad/configs/stage1_track_map/tiny_track_map_b2d.py b/adzoo/uniad/configs/stage1_track_map/tiny_track_map_b2d.py
@@ -27,6 +27,7 @@
     "vehicle.diamondback.century": 'bicycle',
     "vehicle.gazelle.omafiets": 'bicycle',
     # car
+    "vehicle.audi.etron": 'car',
     "vehicle.chevrolet.impala": 'car',
     "vehicle.dodge.charger_2020": 'car',
     "vehicle.dodge.charger_police": 'car',
@@ -82,19 +83,30 @@
 
     #=================pedestrian==============
     "walker.pedestrian.0001": 'pedestrian',
+    "walker.pedestrian.0003": 'pedestrian',
     "walker.pedestrian.0004": 'pedestrian',
     "walker.pedestrian.0005": 'pedestrian',
     "walker.pedestrian.0007": 'pedestrian',
+    "walker.pedestrian.0010": 'pedestrian',
     "walker.pedestrian.0013": 'pedestrian',
     "walker.pedestrian.0014": 'pedestrian',
+    "walker.pedestrian.0015": 'pedestrian',
+    "walker.pedestrian.0016": 'pedestrian',
     "walker.pedestrian.0017": 'pedestrian',
     "walker.pedestrian.0018": 'pedestrian',
     "walker.pedestrian.0019": 'pedestrian',
     "walker.pedestrian.0020": 'pedestrian',
+    "walker.pedestrian.0021": 'pedestrian',
     "walker.pedestrian.0022": 'pedestrian',
     "walker.pedestrian.0025": 'pedestrian',
+    "walker.pedestrian.0027": 'pedestrian',
+    "walker.pedestrian.0030": 'pedestrian',
+    "walker.pedestrian.0031": 'pedestrian',
+    "walker.pedestrian.0032": 'pedestrian',
+    "walker.pedestrian.0034": 'pedestrian',
     "walker.pedestrian.0035": 'pedestrian',
     "walker.pedestrian.0041": 'pedestrian',
+    "walker.pedestrian.0042": 'pedestrian',
     "walker.pedestrian.0046": 'pedestrian',
     "walker.pedestrian.0047": 'pedestrian',
 

diff --git a/adzoo/uniad/configs/stage2_e2e/base_e2e_b2d.py b/adzoo/uniad/configs/stage2_e2e/base_e2e_b2d.py
@@ -10,13 +10,15 @@
 voxel_size = [0.2, 0.2, 8]
 patch_size = [102.4, 102.4]
 img_norm_cfg = dict(mean=[103.530, 116.280, 123.675], std=[1.0, 1.0, 1.0], to_rgb=False)
+
 NameMapping = {
     #=================vehicle=================
     # bicycle
     'vehicle.bh.crossbike': 'bicycle',
     "vehicle.diamondback.century": 'bicycle',
     "vehicle.gazelle.omafiets": 'bicycle',
     # car
+    "vehicle.audi.etron": 'car',
     "vehicle.chevrolet.impala": 'car',
     "vehicle.dodge.charger_2020": 'car',
     "vehicle.dodge.charger_police": 'car',
@@ -72,19 +74,30 @@
 
     #=================pedestrian==============
     "walker.pedestrian.0001": 'pedestrian',
+    "walker.pedestrian.0003": 'pedestrian',
     "walker.pedestrian.0004": 'pedestrian',
     "walker.pedestrian.0005": 'pedestrian',
     "walker.pedestrian.0007": 'pedestrian',
+    "walker.pedestrian.0010": 'pedestrian',
     "walker.pedestrian.0013": 'pedestrian',
     "walker.pedestrian.0014": 'pedestrian',
+    "walker.pedestrian.0015": 'pedestrian',
+    "walker.pedestrian.0016": 'pedestrian',
     "walker.pedestrian.0017": 'pedestrian',
     "walker.pedestrian.0018": 'pedestrian',
     "walker.pedestrian.0019": 'pedestrian',
     "walker.pedestrian.0020": 'pedestrian',
+    "walker.pedestrian.0021": 'pedestrian',
     "walker.pedestrian.0022": 'pedestrian',
     "walker.pedestrian.0025": 'pedestrian',
+    "walker.pedestrian.0027": 'pedestrian',
+    "walker.pedestrian.0030": 'pedestrian',
+    "walker.pedestrian.0031": 'pedestrian',
+    "walker.pedestrian.0032": 'pedestrian',
+    "walker.pedestrian.0034": 'pedestrian',
     "walker.pedestrian.0035": 'pedestrian',
     "walker.pedestrian.0041": 'pedestrian',
+    "walker.pedestrian.0042": 'pedestrian',
     "walker.pedestrian.0046": 'pedestrian',
     "walker.pedestrian.0047": 'pedestrian',
 

diff --git a/adzoo/uniad/configs/stage2_e2e/tiny_e2e_b2d.py b/adzoo/uniad/configs/stage2_e2e/tiny_e2e_b2d.py
@@ -10,13 +10,15 @@
 voxel_size = [0.2, 0.2, 8]
 patch_size = [102.4, 102.4]
 img_norm_cfg = dict(mean=[103.530, 116.280, 123.675], std=[1.0, 1.0, 1.0], to_rgb=False)
+
 NameMapping = {
     #=================vehicle=================
     # bicycle
     'vehicle.bh.crossbike': 'bicycle',
     "vehicle.diamondback.century": 'bicycle',
     "vehicle.gazelle.omafiets": 'bicycle',
     # car
+    "vehicle.audi.etron": 'car',
     "vehicle.chevrolet.impala": 'car',
     "vehicle.dodge.charger_2020": 'car',
     "vehicle.dodge.charger_police": 'car',
@@ -72,19 +74,30 @@
 
     #=================pedestrian==============
     "walker.pedestrian.0001": 'pedestrian',
+    "walker.pedestrian.0003": 'pedestrian',
     "walker.pedestrian.0004": 'pedestrian',
     "walker.pedestrian.0005": 'pedestrian',
     "walker.pedestrian.0007": 'pedestrian',
+    "walker.pedestrian.0010": 'pedestrian',
     "walker.pedestrian.0013": 'pedestrian',
     "walker.pedestrian.0014": 'pedestrian',
+    "walker.pedestrian.0015": 'pedestrian',
+    "walker.pedestrian.0016": 'pedestrian',
     "walker.pedestrian.0017": 'pedestrian',
     "walker.pedestrian.0018": 'pedestrian',
     "walker.pedestrian.0019": 'pedestrian',
     "walker.pedestrian.0020": 'pedestrian',
+    "walker.pedestrian.0021": 'pedestrian',
     "walker.pedestrian.0022": 'pedestrian',
     "walker.pedestrian.0025": 'pedestrian',
+    "walker.pedestrian.0027": 'pedestrian',
+    "walker.pedestrian.0030": 'pedestrian',
+    "walker.pedestrian.0031": 'pedestrian',
+    "walker.pedestrian.0032": 'pedestrian',
+    "walker.pedestrian.0034": 'pedestrian',
     "walker.pedestrian.0035": 'pedestrian',
     "walker.pedestrian.0041": 'pedestrian',
+    "walker.pedestrian.0042": 'pedestrian',
     "walker.pedestrian.0046": 'pedestrian',
     "walker.pedestrian.0047": 'pedestrian',
 

diff --git a/adzoo/vad/configs/VAD/VAD_base_e2e_b2d.py b/adzoo/vad/configs/VAD/VAD_base_e2e_b2d.py
@@ -20,6 +20,7 @@
     "vehicle.diamondback.century": 'bicycle',
     "vehicle.gazelle.omafiets": 'bicycle',
     # car
+    "vehicle.audi.etron": 'car',
     "vehicle.chevrolet.impala": 'car',
     "vehicle.dodge.charger_2020": 'car',
     "vehicle.dodge.charger_police": 'car',
@@ -75,19 +76,30 @@
 
     #=================pedestrian==============
     "walker.pedestrian.0001": 'pedestrian',
+    "walker.pedestrian.0003": 'pedestrian',
     "walker.pedestrian.0004": 'pedestrian',
     "walker.pedestrian.0005": 'pedestrian',
     "walker.pedestrian.0007": 'pedestrian',
+    "walker.pedestrian.0010": 'pedestrian',
     "walker.pedestrian.0013": 'pedestrian',
     "walker.pedestrian.0014": 'pedestrian',
+    "walker.pedestrian.0015": 'pedestrian',
+    "walker.pedestrian.0016": 'pedestrian',
     "walker.pedestrian.0017": 'pedestrian',
     "walker.pedestrian.0018": 'pedestrian',
     "walker.pedestrian.0019": 'pedestrian',
     "walker.pedestrian.0020": 'pedestrian',
+    "walker.pedestrian.0021": 'pedestrian',
     "walker.pedestrian.0022": 'pedestrian',
     "walker.pedestrian.0025": 'pedestrian',
+    "walker.pedestrian.0027": 'pedestrian',
+    "walker.pedestrian.0030": 'pedestrian',
+    "walker.pedestrian.0031": 'pedestrian',
+    "walker.pedestrian.0032": 'pedestrian',
+    "walker.pedestrian.0034": 'pedestrian',
     "walker.pedestrian.0035": 'pedestrian',
     "walker.pedestrian.0041": 'pedestrian',
+    "walker.pedestrian.0042": 'pedestrian',
     "walker.pedestrian.0046": 'pedestrian',
     "walker.pedestrian.0047": 'pedestrian',