fix num_cls

triple-Mu · Apr 15, 2024 · 3659189 · 3659189
1 parent b91ba07
commit 3659189
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/configs/rotated_rtmdet/_base_/comp_rr.py b/configs/rotated_rtmdet/_base_/comp_rr.py
@@ -2,6 +2,12 @@
 dataset_type = 'DOTADataset'
 data_root = 'data/DOTA/'
 
+metainfo = {
+    'classes': ('A', 'B', 'C', 'D', 'E', 'F'),
+    'palette': [(165, 42, 42), (189, 183, 107), (0, 255, 0), (255, 0, 0),
+                (138, 43, 226), (255, 128, 0)]
+}
+
 file_client_args = dict(backend='disk')
 
 train_pipeline = [
@@ -58,6 +64,7 @@
     dataset=dict(
         type=dataset_type,
         data_root=data_root,
+        metainfo=metainfo,
         ann_file='train/annfiles/',
         data_prefix=dict(img_path='train/images/'),
         img_shape=(1024, 1024),
@@ -72,6 +79,7 @@
     dataset=dict(
         type=dataset_type,
         data_root=data_root,
+        metainfo=metainfo,
         ann_file='val/annfiles/',
         data_prefix=dict(img_path='val/images/'),
         img_shape=(1024, 1024),

diff --git a/configs/rotated_rtmdet/rotated_rtmdet_l-3x-dota-comp.py b/configs/rotated_rtmdet/rotated_rtmdet_l-3x-dota-comp.py
@@ -35,7 +35,7 @@
         act_cfg=dict(type='SiLU')),
     bbox_head=dict(
         type='RotatedRTMDetSepBNHead',
-        num_classes=15,
+        num_classes=6,
         in_channels=256,
         stacked_convs=2,
         feat_channels=256,