VOC class order

Adamdad · Mar 30, 2023 · 1fa6477 · 1fa6477
1 parent a2d03ee
commit 1fa6477
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 7 deletions.
diff --git a/configs/baseline/mean_teacher_retinanet_r50_fpn_voc0712_72k.py b/configs/baseline/mean_teacher_retinanet_r50_fpn_voc0712_72k.py
@@ -254,9 +254,9 @@
     ),
 ]
 
-CLASSES = ('aeroplane', 'bicycle', 'bird', 'boat', 'bottle', 'bus', 'car',
-           'cat', 'chair', 'cow', 'diningtable', 'dog', 'horse', 'motorbike',
-           'person', 'pottedplant', 'sheep', 'sofa', 'train', 'tvmonitor')
+CLASSES = ('train', 'car', 'cat', 'chair', 'person', 'diningtable', 'sofa', 
+           'horse', 'bicycle', 'bird', 'cow', 'aeroplane', 'tvmonitor', 
+           'bottle', 'pottedplant', 'boat', 'sheep', 'bus', 'motorbike', 'dog')
 data = dict(
     samples_per_gpu=5,
     workers_per_gpu=5,

diff --git a/configs/consistent-teacher/consistent_teacher_r50_fpn_voc0712_72k.py b/configs/consistent-teacher/consistent_teacher_r50_fpn_voc0712_72k.py
@@ -252,10 +252,6 @@
 ]
 
 
-# CLASSES = ('aeroplane', 'bicycle', 'bird', 'boat', 'bottle', 'bus', 'car',
-#            'cat', 'chair', 'cow', 'diningtable', 'dog', 'horse', 'motorbike',
-#            'person', 'pottedplant', 'sheep', 'sofa', 'train', 'tvmonitor')
-
 CLASSES = ('train', 'car', 'cat', 'chair', 'person', 'diningtable', 'sofa', 
            'horse', 'bicycle', 'bird', 'cow', 'aeroplane', 'tvmonitor', 
            'bottle', 'pottedplant', 'boat', 'sheep', 'bus', 'motorbike', 'dog')