update config files

configs/config_rodnet_cdc_win16.py

@@ -24,7 +24,7 @@ dataset_cfg = dict(
 
 model_cfg = dict(
     type='CDC',
-    name='rodnet-cdc-win16-wobg',
+    name='rodnet-cdc-win16',
     max_dets=20,
     peak_thres=0.3,
     ols_thres=0.3,

configs/config_rodnet_cdc_win16_mini.py

@@ -24,7 +24,7 @@ dataset_cfg = dict(
 
 model_cfg = dict(
     type='CDC',
-    name='rodnet-cdc-win16-wobg',
+    name='rodnet-cdc-win16',
     max_dets=20,
     peak_thres=0.3,
     ols_thres=0.3,

configs/config_rodnet_cdc_win16_mnet.py

+dataset_cfg = dict(
+    dataset_name='ROD2021',
+    base_root="/mnt/disk1/CRUW/ROD2021",
+    data_root="/mnt/disk1/CRUW/ROD2021/sequences",
+    anno_root="/mnt/disk1/CRUW/ROD2021/annotations",
+    anno_ext='.txt',
+    train=dict(
+        subdir='train',
+        # seqs=[],  # can choose from the subdir folder
+    ),
+    valid=dict(
+        subdir='valid',
+        seqs=[],
+    ),
+    test=dict(
+        subdir='test',
+        # seqs=[],  # can choose from the subdir folder
+    ),
+    demo=dict(
+        subdir='demo',
+        seqs=[],
+    ),
+)
+
+model_cfg = dict(
+    type='CDC',
+    name='rodnet-cdc-win16-mnet',
+    max_dets=20,
+    peak_thres=0.3,
+    ols_thres=0.3,
+    mnet_cfg=(4, 32),
+)
+
+confmap_cfg = dict(
+    confmap_sigmas={
+        'pedestrian': 15,
+        'cyclist': 20,
+        'car': 30,
+        # 'van': 40,
+        # 'truck': 50,
+    },
+    confmap_sigmas_interval={
+        'pedestrian': [5, 15],
+        'cyclist': [8, 20],
+        'car': [10, 30],
+        # 'van': [15, 40],
+        # 'truck': [20, 50],
+    },
+    confmap_length={
+        'pedestrian': 1,
+        'cyclist': 2,
+        'car': 3,
+        # 'van': 4,
+        # 'truck': 5,
+    }
+)
+
+train_cfg = dict(
+    n_epoch=100,
+    batch_size=4,
+    lr=0.00001,
+    lr_step=5,  # lr will decrease 10 times after lr_step epoches
+    win_size=16,
+    train_step=1,
+    train_stride=4,
+    log_step=100,
+    save_step=10000,
+)
+test_cfg = dict(
+    test_step=1,
+    test_stride=8,
+    rr_min=1.0,  # min radar range
+    rr_max=20.0,  # max radar range
+    ra_min=-60.0,  # min radar angle
+    ra_max=60.0,  # max radar angle
+)

configs/config_rodnet_v2_cdc_win16.py → configs/config_rodnet_cdcv2_win16_mnet_dcn.py

@@ -24,7 +24,7 @@ dataset_cfg = dict(
 
 model_cfg = dict(
     type='CDCv2',
-    name='rodnet-cdc-v2-win16-wobg',
+    name='rodnet-cdcv2-win16-mnet-dcn',
     max_dets=20,
     peak_thres=0.3,
     ols_thres=0.3,

configs/config_rodnet_hg1_win16.py

@@ -24,7 +24,7 @@ dataset_cfg = dict(
 
 model_cfg = dict(
     type='HG',
-    name='rodnet-hg1-win16-wobg',
+    name='rodnet-hg1-win16',
     max_dets=20,
     peak_thres=0.4,
     ols_thres=0.3,

configs/config_rodnet_hg1_win16_mnet.py

+dataset_cfg = dict(
+    dataset_name='ROD2021',
+    base_root="/mnt/disk2/CRUW/ROD2021",
+    data_root="/mnt/disk2/CRUW/ROD2021/sequences",
+    anno_root="/mnt/disk2/CRUW/ROD2021/annotations",
+    anno_ext='.txt',
+    train=dict(
+        subdir='train',
+        # seqs=[],  # can choose from the subdir folder
+    ),
+    valid=dict(
+        subdir='valid',
+        seqs=[],
+    ),
+    test=dict(
+        subdir='test',
+        # seqs=[],  # can choose from the subdir folder
+    ),
+    demo=dict(
+        subdir='demo',
+        seqs=[],
+    ),
+)
+
+model_cfg = dict(
+    type='HG',
+    name='rodnet-hg1-win16-mnet',
+    max_dets=20,
+    peak_thres=0.4,
+    ols_thres=0.3,
+    stacked_num=1,
+    mnet_cfg=(4, 32),
+)
+
+confmap_cfg = dict(
+    confmap_sigmas={
+        'pedestrian': 15,
+        'cyclist': 20,
+        'car': 30,
+        # 'van': 40,
+        # 'truck': 50,
+    },
+    confmap_sigmas_interval={
+        'pedestrian': [5, 15],
+        'cyclist': [8, 20],
+        'car': [10, 30],
+        # 'van': [15, 40],
+        # 'truck': [20, 50],
+    },
+    confmap_length={
+        'pedestrian': 1,
+        'cyclist': 2,
+        'car': 3,
+        # 'van': 4,
+        # 'truck': 5,
+    }
+)
+
+train_cfg = dict(
+    n_epoch=50,
+    batch_size=4,
+    lr=0.00001,
+    lr_step=5,  # lr will decrease 10 times after lr_step epoches
+    win_size=16,
+    train_step=1,
+    train_stride=4,
+    log_step=100,
+    save_step=10000,
+)
+test_cfg = dict(
+    test_step=1,
+    test_stride=8,
+    rr_min=1.0,  # min radar range
+    rr_max=20.0,  # max radar range
+    ra_min=-60.0,  # min radar angle
+    ra_max=60.0,  # max radar angle
+)

configs/config_rodnet_v2_hg1_win16.py → configs/config_rodnet_hg1v2_win16_mnet_dcn.py

@@ -24,7 +24,7 @@ dataset_cfg = dict(
 
 model_cfg = dict(
     type='HGv2',
-    name='rodnet-hg1-v2-win16-wobg',
+    name='rodnet-hg1v2-win16-mnet-dcn',
     max_dets=20,
     peak_thres=0.4,
     ols_thres=0.3,

configs/config_rodnet_hg1wi_win16.py

@@ -24,7 +24,7 @@ dataset_cfg = dict(
 
 model_cfg = dict(
     type='HGwI',
-    name='rodnet-hg1wi-win16-wobg',
+    name='rodnet-hg1wi-win16',
     max_dets=20,
     peak_thres=0.3,
     ols_thres=0.3,

configs/config_rodnet_hg1wi_win16_mnet.py

+dataset_cfg = dict(
+    dataset_name='ROD2021',
+    base_root="/mnt/disk1/CRUW/ROD2021",
+    data_root="/mnt/disk1/CRUW/ROD2021/sequences",
+    anno_root="/mnt/disk1/CRUW/ROD2021/annotations",
+    anno_ext='.txt',
+    train=dict(
+        subdir='train',
+        # seqs=[],  # can choose from the subdir folder
+    ),
+    valid=dict(
+        subdir='valid',
+        # seqs=[],
+    ),
+    test=dict(
+        subdir='test',
+        # seqs=[],  # can choose from the subdir folder
+    ),
+    demo=dict(
+        subdir='demo',
+        seqs=[],
+    ),
+)
+
+model_cfg = dict(
+    type='HGwI',
+    name='rodnet-hg1wi-win16-mnet',
+    max_dets=20,
+    peak_thres=0.3,
+    ols_thres=0.3,
+    stacked_num=1,
+    mnet_cfg=(4, 32),
+)
+
+confmap_cfg = dict(
+    confmap_sigmas={
+        'pedestrian': 15,
+        'cyclist': 20,
+        'car': 30,
+        # 'van': 40,
+        # 'truck': 50,
+    },
+    confmap_sigmas_interval={
+        'pedestrian': [5, 15],
+        'cyclist': [8, 20],
+        'car': [10, 30],
+        # 'van': [15, 40],
+        # 'truck': [20, 50],
+    },
+    confmap_length={
+        'pedestrian': 1,
+        'cyclist': 2,
+        'car': 3,
+        # 'van': 4,
+        # 'truck': 5,
+    }
+)
+
+train_cfg = dict(
+    n_epoch=100,
+    batch_size=2,
+    lr=0.00001,
+    lr_step=5,  # lr will decrease 10 times after lr_step epoches
+    win_size=16,
+    train_step=1,
+    train_stride=4,
+    log_step=100,
+    save_step=10000,
+)
+test_cfg = dict(
+    test_step=1,
+    test_stride=8,
+    rr_min=1.0,  # min radar range
+    rr_max=20.0,  # max radar range
+    ra_min=-60.0,  # min radar angle
+    ra_max=60.0,  # max radar angle
+)

configs/config_rodnet_v2_hg1wi_win16.py → configs/config_rodnet_hg1wiv2_win16_mnet_dcn.py

@@ -24,7 +24,7 @@ dataset_cfg = dict(
 
 model_cfg = dict(
     type='HGwIv2',
-    name='rodnet-hg1wi-v2-win16-wobg',
+    name='rodnet-hg1wiv2-win16-mnet-dcn',
     max_dets=20,
     peak_thres=0.3,
     ols_thres=0.3,