[Config] rename base dataset terms to {dataset-name}_task_train/test (#1541)

Author: Harold-lkk

configs/textdet/_base_/datasets/ctw1500.py

@@ -1,16 +1,16 @@
-ctw_det_data_root = 'data/det/ctw1500'
+ctw1500_textdet_data_root = 'data/det/ctw1500'
 
-ctw_det_train = dict(
+ctw1500_textdet_train = dict(
     type='OCRDataset',
-    data_root=ctw_det_data_root,
+    data_root=ctw1500_textdet_data_root,
     ann_file='instances_training.json',
     data_prefix=dict(img_path='imgs/'),
     filter_cfg=dict(filter_empty_gt=True, min_size=32),
     pipeline=None)
 
-ctw_det_test = dict(
+ctw1500_textdet_test = dict(
     type='OCRDataset',
-    data_root=ctw_det_data_root,
+    data_root=ctw1500_textdet_data_root,
     ann_file='instances_test.json',
     data_prefix=dict(img_path='imgs/'),
     test_mode=True,

configs/textdet/_base_/datasets/icdar2015.py

@@ -1,16 +1,16 @@
-ic15_det_data_root = 'data/det/icdar2015'
+icdar2015_textdet_data_root = 'data/det/icdar2015'
 
-ic15_det_train = dict(
+icdar2015_textdet_train = dict(
     type='OCRDataset',
-    data_root=ic15_det_data_root,
+    data_root=icdar2015_textdet_data_root,
     ann_file='instances_training.json',
     data_prefix=dict(img_path='imgs/'),
     filter_cfg=dict(filter_empty_gt=True, min_size=32),
     pipeline=None)
 
-ic15_det_test = dict(
+icdar2015_textdet_test = dict(
     type='OCRDataset',
-    data_root=ic15_det_data_root,
+    data_root=icdar2015_textdet_data_root,
     ann_file='instances_test.json',
     data_prefix=dict(img_path='imgs/'),
     test_mode=True,

configs/textdet/_base_/datasets/icdar2017.py

@@ -1,16 +1,16 @@
-ic17_det_data_root = 'data/det/icdar_2017'
+icdar2017_textdet_data_root = 'data/det/icdar_2017'
 
-ic17_det_train = dict(
+icdar2017_textdet_train = dict(
     type='OCRDataset',
-    data_root=ic17_det_data_root,
+    data_root=icdar2017_textdet_data_root,
     ann_file='instances_training.json',
     data_prefix=dict(img_path='imgs/'),
     filter_cfg=dict(filter_empty_gt=True, min_size=32),
     pipeline=None)
 
-ic17_det_test = dict(
+icdar2017_textdet_test = dict(
     type='OCRDataset',
-    data_root=ic17_det_data_root,
+    data_root=icdar2017_textdet_data_root,
     ann_file='instances_test.json',
     data_prefix=dict(img_path='imgs/'),
     test_mode=True,

configs/textdet/_base_/datasets/synthtext.py

@@ -1,16 +1,16 @@
-st_det_data_root = 'data/det/synthtext'
+synthtext_textdet_data_root = 'data/det/synthtext'
 
-st_det_train = dict(
+synthtext_textdet_train = dict(
     type='OCRDataset',
-    data_root=st_det_data_root,
+    data_root=synthtext_textdet_data_root,
     ann_file='instances_training.json',
     data_prefix=dict(img_path='imgs/'),
     filter_cfg=dict(filter_empty_gt=True, min_size=32),
     pipeline=None)
 
-st_det_test = dict(
+synthtext_textdet_test = dict(
     type='OCRDataset',
-    data_root=st_det_data_root,
+    data_root=synthtext_textdet_data_root,
     ann_file='instances_test.json',
     data_prefix=dict(img_path='imgs/'),
     test_mode=True,

configs/textdet/_base_/datasets/totaltext.py

@@ -1,15 +1,15 @@
-tt_det_data_root = 'data/totaltext'
+totaltext_textdet_data_root = 'data/totaltext'
 
-tt_det_train = dict(
+totaltext_textdet_train = dict(
     type='OCRDataset',
-    data_root=tt_det_data_root,
+    data_root=totaltext_textdet_data_root,
     ann_file='textdet_train.json',
     filter_cfg=dict(filter_empty_gt=True, min_size=32),
     pipeline=None)
 
-tt_det_test = dict(
+totaltext_textdet_test = dict(
     type='OCRDataset',
-    data_root=tt_det_data_root,
+    data_root=totaltext_textdet_data_root,
     ann_file='textdet_test.json',
     test_mode=True,
     pipeline=None)

configs/textdet/dbnet/dbnet_resnet18_fpnc_100k_synthtext.py

@@ -6,24 +6,24 @@
 ]
 
 # dataset settings
-st_det_train = _base_.st_det_train
-st_det_train.pipeline = _base_.train_pipeline
-st_det_test = _base_.st_det_test
-st_det_test.pipeline = _base_.test_pipeline
+synthtext_textdet_train = _base_.synthtext_textdet_train
+synthtext_textdet_train.pipeline = _base_.train_pipeline
+synthtext_textdet_test = _base_.synthtext_textdet_test
+synthtext_textdet_test.pipeline = _base_.test_pipeline
 
 train_dataloader = dict(
     batch_size=16,
     num_workers=8,
     persistent_workers=True,
     sampler=dict(type='DefaultSampler', shuffle=True),
-    dataset=st_det_train)
+    dataset=synthtext_textdet_train)
 
 val_dataloader = dict(
     batch_size=1,
     num_workers=4,
     persistent_workers=True,
     sampler=dict(type='DefaultSampler', shuffle=False),
-    dataset=st_det_test)
+    dataset=synthtext_textdet_test)
 
 test_dataloader = val_dataloader
 

configs/textdet/dbnet/dbnet_resnet18_fpnc_1200e_icdar2015.py

@@ -6,24 +6,24 @@
 ]
 
 # dataset settings
-ic15_det_train = _base_.ic15_det_train
-ic15_det_train.pipeline = _base_.train_pipeline
-ic15_det_test = _base_.ic15_det_test
-ic15_det_test.pipeline = _base_.test_pipeline
+icdar2015_textdet_train = _base_.icdar2015_textdet_train
+icdar2015_textdet_train.pipeline = _base_.train_pipeline
+icdar2015_textdet_test = _base_.icdar2015_textdet_test
+icdar2015_textdet_test.pipeline = _base_.test_pipeline
 
 train_dataloader = dict(
     batch_size=16,
     num_workers=8,
     persistent_workers=True,
     sampler=dict(type='DefaultSampler', shuffle=True),
-    dataset=ic15_det_train)
+    dataset=icdar2015_textdet_train)
 
 val_dataloader = dict(
     batch_size=1,
     num_workers=4,
     persistent_workers=True,
     sampler=dict(type='DefaultSampler', shuffle=False),
-    dataset=ic15_det_test)
+    dataset=icdar2015_textdet_test)
 
 test_dataloader = val_dataloader
 

configs/textdet/dbnet/dbnet_resnet50-dcnv2_fpnc_100k_synthtext.py

@@ -6,24 +6,24 @@
 ]
 
 # dataset settings
-st_det_train = _base_.st_det_train
-st_det_train.pipeline = _base_.train_pipeline
-st_det_test = _base_.st_det_test
-st_det_test.pipeline = _base_.test_pipeline
+synthtext_textdet_train = _base_.synthtext_textdet_train
+synthtext_textdet_train.pipeline = _base_.train_pipeline
+synthtext_textdet_test = _base_.synthtext_textdet_test
+synthtext_textdet_test.pipeline = _base_.test_pipeline
 
 train_dataloader = dict(
     batch_size=16,
     num_workers=8,
     persistent_workers=True,
     sampler=dict(type='DefaultSampler', shuffle=True),
-    dataset=st_det_train)
+    dataset=synthtext_textdet_train)
 
 val_dataloader = dict(
     batch_size=1,
     num_workers=4,
     persistent_workers=True,
     sampler=dict(type='DefaultSampler', shuffle=False),
-    dataset=st_det_test)
+    dataset=synthtext_textdet_test)
 
 test_dataloader = val_dataloader
 

configs/textdet/dbnet/dbnet_resnet50-dcnv2_fpnc_1200e_icdar2015.py

@@ -9,24 +9,24 @@
 load_from = 'https://download.openmmlab.com/mmocr/textdet/dbnet/tmp_1.0_pretrain/dbnet_r50dcnv2_fpnc_sbn_2e_synthtext_20210325-ed322016.pth'  # noqa
 
 # dataset settings
-ic15_det_train = _base_.ic15_det_train
-ic15_det_train.pipeline = _base_.train_pipeline
-ic15_det_test = _base_.ic15_det_test
-ic15_det_test.pipeline = _base_.test_pipeline
+icdar2015_textdet_train = _base_.icdar2015_textdet_train
+icdar2015_textdet_train.pipeline = _base_.train_pipeline
+icdar2015_textdet_test = _base_.icdar2015_textdet_test
+icdar2015_textdet_test.pipeline = _base_.test_pipeline
 
 train_dataloader = dict(
     batch_size=16,
     num_workers=8,
     persistent_workers=True,
     sampler=dict(type='DefaultSampler', shuffle=True),
-    dataset=ic15_det_train)
+    dataset=icdar2015_textdet_train)
 
 val_dataloader = dict(
     batch_size=1,
     num_workers=4,
     persistent_workers=True,
     sampler=dict(type='DefaultSampler', shuffle=False),
-    dataset=ic15_det_test)
+    dataset=icdar2015_textdet_test)
 
 test_dataloader = val_dataloader
 

configs/textdet/dbnetpp/dbnetpp_resnet50-dcnv2_fpnc_100k_synthtext.py

@@ -6,8 +6,8 @@
 ]
 
 # dataset settings
-train_list = [_base_.st_det_train]
-test_list = [_base_.st_det_test]
+train_list = [_base_.synthtext_textdet_train]
+test_list = [_base_.synthtext_textdet_test]
 
 train_dataloader = dict(
     batch_size=16,

configs/textdet/dbnetpp/dbnetpp_resnet50-dcnv2_fpnc_1200e_icdar2015.py

@@ -8,8 +8,8 @@
 load_from = 'https://download.openmmlab.com/mmocr/textdet/dbnetpp/tmp_1.0_pretrain/dbnetpp_r50dcnv2_fpnc_100k_iter_synthtext-20220502-352fec8a.pth'  # noqa
 
 # dataset settings
-train_list = [_base_.ic15_det_train]
-test_list = [_base_.ic15_det_test]
+train_list = [_base_.icdar2015_textdet_train]
+test_list = [_base_.icdar2015_textdet_test]
 
 train_dataloader = dict(
     batch_size=16,

configs/textdet/drrg/drrg_resnet50_fpn-unet_1200e_ctw1500.py

@@ -6,24 +6,24 @@
 ]
 
 # dataset settings
-ctw_det_train = _base_.ctw_det_train
-ctw_det_train.pipeline = _base_.train_pipeline
-ctw_det_test = _base_.ctw_det_test
-ctw_det_test.pipeline = _base_.test_pipeline
+ctw1500_textdet_train = _base_.ctw1500_textdet_train
+ctw1500_textdet_train.pipeline = _base_.train_pipeline
+ctw1500_textdet_test = _base_.ctw1500_textdet_test
+ctw1500_textdet_test.pipeline = _base_.test_pipeline
 
 train_dataloader = dict(
     batch_size=4,
     num_workers=4,
     persistent_workers=True,
     sampler=dict(type='DefaultSampler', shuffle=True),
-    dataset=ctw_det_train)
+    dataset=ctw1500_textdet_train)
 
 val_dataloader = dict(
     batch_size=1,
     num_workers=1,
     persistent_workers=True,
     sampler=dict(type='DefaultSampler', shuffle=False),
-    dataset=ctw_det_test)
+    dataset=ctw1500_textdet_test)
 
 test_dataloader = val_dataloader
 

configs/textdet/fcenet/fcenet_resnet50-dcnv2_fpn_1500e_ctw1500.py

@@ -14,8 +14,8 @@
 
 file_client_args = dict(backend='disk')
 # dataset settings
-ctw_det_train = _base_.ctw_det_train
-ctw_det_test = _base_.ctw_det_test
+ctw1500_textdet_train = _base_.ctw1500_textdet_train
+ctw1500_textdet_test = _base_.ctw1500_textdet_test
 
 # test pipeline for CTW1500
 ctw_test_pipeline = [
@@ -36,22 +36,22 @@
         meta_keys=('img_path', 'ori_shape', 'img_shape', 'scale_factor'))
 ]
 
-ctw_det_train.pipeline = _base_.train_pipeline
-ctw_det_test.pipeline = ctw_test_pipeline
+ctw1500_textdet_train.pipeline = _base_.train_pipeline
+ctw1500_textdet_test.pipeline = ctw_test_pipeline
 
 train_dataloader = dict(
     batch_size=8,
     num_workers=4,
     persistent_workers=True,
     sampler=dict(type='DefaultSampler', shuffle=True),
-    dataset=ctw_det_train)
+    dataset=ctw1500_textdet_train)
 
 val_dataloader = dict(
     batch_size=1,
     num_workers=1,
     persistent_workers=True,
     sampler=dict(type='DefaultSampler', shuffle=False),
-    dataset=ctw_det_test)
+    dataset=ctw1500_textdet_test)
 
 test_dataloader = val_dataloader
 

configs/textdet/fcenet/fcenet_resnet50_fpn_1500e_icdar2015.py

@@ -13,24 +13,24 @@
 ]
 
 # dataset settings
-ic15_det_train = _base_.ic15_det_train
-ic15_det_test = _base_.ic15_det_test
-ic15_det_train.pipeline = _base_.train_pipeline
-ic15_det_test.pipeline = _base_.test_pipeline
+icdar2015_textdet_train = _base_.icdar2015_textdet_train
+icdar2015_textdet_test = _base_.icdar2015_textdet_test
+icdar2015_textdet_train.pipeline = _base_.train_pipeline
+icdar2015_textdet_test.pipeline = _base_.test_pipeline
 
 train_dataloader = dict(
     batch_size=8,
     num_workers=4,
     persistent_workers=True,
     sampler=dict(type='DefaultSampler', shuffle=True),
-    dataset=ic15_det_train)
+    dataset=icdar2015_textdet_train)
 
 val_dataloader = dict(
     batch_size=1,
     num_workers=1,
     persistent_workers=True,
     sampler=dict(type='DefaultSampler', shuffle=False),
-    dataset=ic15_det_test)
+    dataset=icdar2015_textdet_test)
 
 test_dataloader = val_dataloader
 

configs/textdet/maskrcnn/mask-rcnn_resnet50_fpn_160e_ctw1500.py

@@ -15,8 +15,8 @@
 ]
 
 # dataset settings
-ctw_det_train = _base_.ctw_det_train
-ctw_det_test = _base_.ctw_det_test
+ctw1500_textdet_train = _base_.ctw1500_textdet_train
+ctw1500_textdet_test = _base_.ctw1500_textdet_test
 
 # test pipeline for CTW1500
 ctw_test_pipeline = [
@@ -37,22 +37,22 @@
         meta_keys=('img_path', 'ori_shape', 'img_shape', 'scale_factor'))
 ]
 
-ctw_det_train.pipeline = _base_.train_pipeline
-ctw_det_test.pipeline = ctw_test_pipeline
+ctw1500_textdet_train.pipeline = _base_.train_pipeline
+ctw1500_textdet_test.pipeline = ctw_test_pipeline
 
 train_dataloader = dict(
     batch_size=8,
     num_workers=4,
     persistent_workers=True,
     sampler=dict(type='DefaultSampler', shuffle=True),
-    dataset=ctw_det_train)
+    dataset=ctw1500_textdet_train)
 
 val_dataloader = dict(
     batch_size=1,
     num_workers=1,
     persistent_workers=True,
     sampler=dict(type='DefaultSampler', shuffle=False),
-    dataset=ctw_det_test)
+    dataset=ctw1500_textdet_test)
 
 test_dataloader = val_dataloader