mmocr/configs/_base_/recog_datasets/academic_test.py
2022-07-21 10:58:04 +08:00

70 lines
1.7 KiB
Python

# Text Recognition Testing set, including:
# Regular Datasets: IIIT5K, SVT, IC13
# Irregular Datasets: IC15, SVTP, CT80
test_root = 'data/rec'
test_img_prefix1 = 'IIIT5K/'
test_img_prefix2 = 'svt/'
test_img_prefix3 = 'icdar_2013/Challenge2_Test_Task3_Images/'
test_img_prefix4 = 'icdar_2015/ch4_test_word_images_gt'
test_img_prefix5 = 'svtp/'
test_img_prefix6 = 'ct80/'
test_ann_file1 = 'IIIT5K/test_label.json'
test_ann_file2 = 'svt/test_label.json'
test_ann_file3 = 'icdar_2013/test_label.json'
test_ann_file4 = 'icdar_2015/test_label.json'
test_ann_file5 = 'svtp/test_label.json'
test_ann_file6 = 'ct80/test_label.json'
IIIT5K = dict(
type='OCRDataset',
data_root=test_root,
data_prefix=dict(img_path=test_img_prefix1),
ann_file=test_ann_file1,
test_mode=True,
pipeline=None)
SVT = dict(
type='OCRDataset',
data_root=test_root,
data_prefix=dict(img_path=test_img_prefix2),
ann_file=test_ann_file2,
test_mode=True,
pipeline=None)
IC13 = dict(
type='OCRDataset',
data_root=test_root,
data_prefix=dict(img_path=test_img_prefix3),
ann_file=test_ann_file3,
test_mode=True,
pipeline=None)
IC15 = dict(
type='OCRDataset',
data_root=test_root,
data_prefix=dict(img_path=test_img_prefix4),
ann_file=test_ann_file4,
test_mode=True,
pipeline=None)
SVTP = dict(
type='OCRDataset',
data_root=test_root,
data_prefix=dict(img_path=test_img_prefix5),
ann_file=test_ann_file5,
test_mode=True,
pipeline=None)
CUTE80 = dict(
type='OCRDataset',
data_root=test_root,
data_prefix=dict(img_path=test_img_prefix6),
ann_file=test_ann_file6,
test_mode=True,
pipeline=None)
test_list = [IIIT5K, SVT, IC13, IC15, SVTP, CUTE80]