Update config

gist-ailab · Dec 19, 2024 · b544318 · b544318
1 parent d797652
commit b544318
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/...50_fpn_2x_flair-adas-rgbt-v4-traintest.py → ...st_rgbt/AttNet_r50_fpn_2x_kaistRgbtAll.py b/...50_fpn_2x_flair-adas-rgbt-v4-traintest.py → ...st_rgbt/AttNet_r50_fpn_2x_kaistRgbtAll.py
@@ -1,4 +1,5 @@
 import os
+# import mmdet
 # dataset settings
 _base_ = [
     '../../../../configs/_base_/models/faster-rcnn_r50_fpn.py',
@@ -7,11 +8,19 @@
     '../../../../configs/_base_/default_runtime.py'
 ]
 
+# _base_ = [
+#     'mmdet::_base_/schedules/schedule_1x.py',
+#     'mmdet::_base_/datasets/coco_detection.py',
+#     'mmdet::_base_/schedules/schedule_2x.py',
+#     'mmdet::_base_/default_runtime.py'
+# ]
 dataset_type = 'KaistRgbtCocoDataset'
 data_root = '/media/ailab/HDD1/Workspace/src/Project/Drone24/detection/SOTA/CAFF-DETR/CAFF-DINO/data/kaist_coco'
 backend_args = None
 
 classes = ('person', 'people', 'cyclist' )
+optimizer = dict(type='SGD', lr=0.01, momentum=0.9, weight_decay=0.0001)
+
 
 model = dict(
     type = 'MultiModalFasterRCNN',
@@ -44,7 +53,7 @@
 ]
 
 train_dataloader = dict(
-    batch_size=2,
+    batch_size=4,
     num_workers=2,
     persistent_workers=True,
     sampler=dict(type='CustomSampler', shuffle=True),
@@ -70,7 +79,7 @@
         type=dataset_type,
         data_root=data_root,
         ann_file='annotations/test-all-01.txt.json',
-        data_prefix=dict(img='         '),
+        data_prefix=dict(img=''),
         test_mode=True,
         pipeline=test_pipeline,
         backend_args=backend_args))

diff --git a/tools/train_debug.py b/tools/train_debug.py
@@ -13,7 +13,7 @@
 
 def parse_args():
     parser = argparse.ArgumentParser(description='Train a detector')
-    parser.add_argument('--config',default='/media/ailab/HDD1/Workspace/src/Project/Drone24/detection/drone-mmdetection-jm/custom_configs/Project_Drone/Paper/kaist_rgbt/AttNet_r50_fpn_2x_flair-adas-rgbt-v4-traintest.py', help='train config file path')
+    parser.add_argument('--config',default='/media/ailab/HDD1/Workspace/src/Project/Drone24/detection/drone-mmdetection-jm/custom_configs/Project_Drone/Paper/kaist_rgbt/AttNet_r50_fpn_2x_kaistRgbtAll.py', help='train config file path')
     parser.add_argument('--work-dir', help='the dir to save logs and models')
     parser.add_argument(
         '--amp',