-
Notifications
You must be signed in to change notification settings - Fork 16
/
icdar2017_traval_taiji.py
36 lines (30 loc) · 1022 Bytes
/
icdar2017_traval_taiji.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
dataset_type = 'OCRCLIPDataset'
data_root = '/apdcephfs/share_887471/interns/v_fisherwyu/data/mmocr_det_data/icdar2017'
train = dict(
type=dataset_type,
ann_file=f'{data_root}/instances_training.json',
img_prefix=f'{data_root}/imgs',
data_name='icdar2017_train',
pipeline=None)
train2 = dict(
type=dataset_type,
ann_file=f'{data_root}/instances_val.json',
img_prefix=f'{data_root}/imgs',
data_name='icdar2017_val',
pipeline=None)
# mlt19_data_root = '/apdcephfs/share_887471/interns/v_fisherwyu/data/mmocr_det_data/mlt2019'
# test_mlt19 = dict(
# type=dataset_type,
# ann_file=f'{mlt19_data_root}/instances_training.json',
# img_prefix=f'{mlt19_data_root}/imgs',
# data_name='mlt2019_train',
# pipeline=None)
test = dict(
type=dataset_type,
ann_file=f'{data_root}/instances_test.json',
img_prefix=f'{data_root}/imgs',
data_name='icdar2017_test',
pipeline=None)
train_list = [train, train2]
# test_list = [test_mlt19]
test_list = [test]