[Fix] Update ST_SA_MJ_train.py (#1117)

Fix a bug that results train3's parameter covering the train1 and train2
pull/1134/head
Mingyu Liu 2022-06-28 14:20:21 +08:00 committed by GitHub
parent fe64040581
commit 9bdc247c0c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 15 additions and 9 deletions

View File

@ -29,14 +29,20 @@ train2 = {key: value for key, value in train1.items()}
train2['img_prefix'] = train_img_prefix2
train2['ann_file'] = train_ann_file2
train3 = {key: value for key, value in train1.items()}
train3['img_prefix'] = train_img_prefix3
train3['ann_file'] = train_ann_file3
train3['loader']['file_format'] = 'txt'
train3['loader']['parser'] = dict(
type='LineStrParser',
keys=['filename', 'text'],
keys_idx=[0, 1],
separator=' ')
train3 = dict(
type='OCRDataset',
img_prefix=train_img_prefix3,
ann_file=train_ann_file3,
loader=dict(
type='AnnFileLoader',
repeat=1,
file_format='txt',
parser=dict(
type='LineStrParser',
keys=['filename', 'text'],
keys_idx=[0, 1],
separator=' ')),
pipeline=None,
test_mode=False)
train_list = [train1, train2, train3]