Spaces:
Runtime error
Runtime error
prefix = 'tests/data/ocr_char_ann_toy_dataset/' | |
train = dict( | |
type='OCRSegDataset', | |
img_prefix=f'{prefix}/imgs', | |
ann_file=f'{prefix}/instances_train.txt', | |
loader=dict( | |
type='HardDiskLoader', | |
repeat=100, | |
parser=dict( | |
type='LineJsonParser', keys=['file_name', 'annotations', 'text'])), | |
pipeline=None, | |
test_mode=True) | |
test = dict( | |
type='OCRDataset', | |
img_prefix=f'{prefix}/imgs', | |
ann_file=f'{prefix}/instances_test.txt', | |
loader=dict( | |
type='HardDiskLoader', | |
repeat=1, | |
parser=dict( | |
type='LineStrParser', | |
keys=['filename', 'text'], | |
keys_idx=[0, 1], | |
separator=' ')), | |
pipeline=None, | |
test_mode=True) | |
train_list = [train] | |
test_list = [test] | |