Spaces:
Sleeping
Sleeping
File size: 955 Bytes
9bf4bd7 |
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 |
num_classes = 26
model = dict(
type='SDMGR',
kie_head=dict(
type='SDMGRHead',
visual_dim=16,
num_classes=num_classes,
module_loss=dict(type='SDMGRModuleLoss'),
postprocessor=dict(type='SDMGRPostProcessor')),
dictionary=dict(
type='Dictionary',
dict_file='{{ fileDirname }}/../../../dicts/sdmgr_dict.txt',
with_padding=True,
with_unknown=True,
unknown_token=None),
)
train_pipeline = [
dict(type='LoadKIEAnnotations'),
dict(type='Resize', scale=(1024, 512), keep_ratio=True),
dict(type='PackKIEInputs')
]
test_pipeline = [
dict(type='LoadKIEAnnotations'),
dict(type='Resize', scale=(1024, 512), keep_ratio=True),
dict(type='PackKIEInputs'),
]
val_evaluator = dict(
type='F1Metric',
mode='macro',
num_classes=num_classes,
ignored_classes=[0, 2, 4, 6, 8, 10, 12, 14, 16, 18, 20, 22, 24, 25])
test_evaluator = val_evaluator
|