mirror of
https://github.com/open-mmlab/mmocr.git
synced 2025-06-03 21:54:47 +08:00
f-str
Signed-off-by: lizz <lizz@sensetime.com>
This commit is contained in:
parent
4f5cde9120
commit
49b52c9d8a
@ -26,7 +26,7 @@ test_pipeline = [
|
|||||||
]
|
]
|
||||||
|
|
||||||
dataset_type = 'KIEDataset'
|
dataset_type = 'KIEDataset'
|
||||||
data_root = 'data/wildreceipt/'
|
data_root = 'data/wildreceipt'
|
||||||
|
|
||||||
loader = dict(
|
loader = dict(
|
||||||
type='HardDiskLoader',
|
type='HardDiskLoader',
|
||||||
@ -37,19 +37,19 @@ loader = dict(
|
|||||||
|
|
||||||
train = dict(
|
train = dict(
|
||||||
type=dataset_type,
|
type=dataset_type,
|
||||||
ann_file=data_root + 'train.txt',
|
ann_file=f'{data_root}/train.txt',
|
||||||
pipeline=train_pipeline,
|
pipeline=train_pipeline,
|
||||||
img_prefix=data_root,
|
img_prefix=data_root,
|
||||||
loader=loader,
|
loader=loader,
|
||||||
dict_file=data_root + 'dict.txt',
|
dict_file=f'{data_root}/dict.txt',
|
||||||
test_mode=False)
|
test_mode=False)
|
||||||
test = dict(
|
test = dict(
|
||||||
type=dataset_type,
|
type=dataset_type,
|
||||||
ann_file=data_root + 'test.txt',
|
ann_file=f'{data_root}/test.txt',
|
||||||
pipeline=test_pipeline,
|
pipeline=test_pipeline,
|
||||||
img_prefix=data_root,
|
img_prefix=data_root,
|
||||||
loader=loader,
|
loader=loader,
|
||||||
dict_file=data_root + 'dict.txt',
|
dict_file=f'{data_root}/dict.txt',
|
||||||
test_mode=True)
|
test_mode=True)
|
||||||
|
|
||||||
data = dict(
|
data = dict(
|
||||||
@ -70,7 +70,7 @@ model = dict(
|
|||||||
visual_modality=False,
|
visual_modality=False,
|
||||||
train_cfg=None,
|
train_cfg=None,
|
||||||
test_cfg=None,
|
test_cfg=None,
|
||||||
class_list=data_root + 'class_list.txt')
|
class_list=f'{data_root}/class_list.txt')
|
||||||
|
|
||||||
optimizer = dict(type='Adam', weight_decay=0.0001)
|
optimizer = dict(type='Adam', weight_decay=0.0001)
|
||||||
optimizer_config = dict(grad_clip=None)
|
optimizer_config = dict(grad_clip=None)
|
||||||
|
@ -26,7 +26,7 @@ test_pipeline = [
|
|||||||
]
|
]
|
||||||
|
|
||||||
dataset_type = 'KIEDataset'
|
dataset_type = 'KIEDataset'
|
||||||
data_root = 'data/wildreceipt/'
|
data_root = 'data/wildreceipt'
|
||||||
|
|
||||||
loader = dict(
|
loader = dict(
|
||||||
type='HardDiskLoader',
|
type='HardDiskLoader',
|
||||||
@ -37,19 +37,19 @@ loader = dict(
|
|||||||
|
|
||||||
train = dict(
|
train = dict(
|
||||||
type=dataset_type,
|
type=dataset_type,
|
||||||
ann_file=data_root + 'train.txt',
|
ann_file=f'{data_root}/train.txt',
|
||||||
pipeline=train_pipeline,
|
pipeline=train_pipeline,
|
||||||
img_prefix=data_root,
|
img_prefix=data_root,
|
||||||
loader=loader,
|
loader=loader,
|
||||||
dict_file=data_root + 'dict.txt',
|
dict_file=f'{data_root}/dict.txt',
|
||||||
test_mode=False)
|
test_mode=False)
|
||||||
test = dict(
|
test = dict(
|
||||||
type=dataset_type,
|
type=dataset_type,
|
||||||
ann_file=data_root + 'test.txt',
|
ann_file=f'{data_root}/test.txt',
|
||||||
pipeline=test_pipeline,
|
pipeline=test_pipeline,
|
||||||
img_prefix=data_root,
|
img_prefix=data_root,
|
||||||
loader=loader,
|
loader=loader,
|
||||||
dict_file=data_root + 'dict.txt',
|
dict_file=f'{data_root}/dict.txt',
|
||||||
test_mode=True)
|
test_mode=True)
|
||||||
|
|
||||||
data = dict(
|
data = dict(
|
||||||
@ -70,7 +70,7 @@ model = dict(
|
|||||||
visual_modality=True,
|
visual_modality=True,
|
||||||
train_cfg=None,
|
train_cfg=None,
|
||||||
test_cfg=None,
|
test_cfg=None,
|
||||||
class_list=data_root + 'class_list.txt')
|
class_list=f'{data_root}/class_list.txt')
|
||||||
|
|
||||||
optimizer = dict(type='Adam', weight_decay=0.0001)
|
optimizer = dict(type='Adam', weight_decay=0.0001)
|
||||||
optimizer_config = dict(grad_clip=None)
|
optimizer_config = dict(grad_clip=None)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user