diff --git a/configs/baseline/mean_teacher_retinanet_r50_fpn_voc0712_72k.py b/configs/baseline/mean_teacher_retinanet_r50_fpn_voc0712_72k.py index 5448cf1..2ae15d4 100644 --- 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 index be218c5..b2c29a8 100644 --- 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')