2022-07-20 11:11:41 +00:00
|
|
|
data_root = 'tests/data/rec_toy_dataset'
|
2022-07-15 11:51:55 +00:00
|
|
|
train_img_prefix = 'imgs/'
|
|
|
|
train_anno_file = 'label.json'
|
2021-11-25 16:27:45 +08:00
|
|
|
|
2022-07-15 11:51:55 +00:00
|
|
|
train_dataset = dict(
|
|
|
|
type='OCRDataset',
|
|
|
|
data_root=data_root,
|
|
|
|
data_prefix=dict(img_path=train_img_prefix),
|
|
|
|
ann_file=train_anno_file,
|
2021-11-25 16:27:45 +08:00
|
|
|
pipeline=None,
|
|
|
|
test_mode=False)
|
|
|
|
|
2022-07-15 11:51:55 +00:00
|
|
|
test_anno_file = f'{data_root}/label.json'
|
|
|
|
test_dataset = dict(
|
|
|
|
type='OCRDataset',
|
|
|
|
data_root=data_root,
|
|
|
|
data_prefix=dict(img_path=train_img_prefix),
|
|
|
|
ann_file=train_anno_file,
|
2021-11-25 16:27:45 +08:00
|
|
|
pipeline=None,
|
|
|
|
test_mode=True)
|
|
|
|
|
2022-07-15 11:51:55 +00:00
|
|
|
train_list = [train_dataset]
|
2021-11-25 16:27:45 +08:00
|
|
|
|
2022-07-15 11:51:55 +00:00
|
|
|
test_list = [test_dataset]
|