mmocr icon indicating copy to clipboard operation
mmocr copied to clipboard

[Metric] metric wrapper

Open Harold-lkk opened this issue 2 years ago • 0 comments


test_list = [    
    _base_.iiit5k_rec_test, _base_.svt_rec_test, _base_.ic13_rec_test,
    _base_.ic15_rec_test, _base_.svtp_rec_test, _base_.cute80_rec_test]

val_dataloader = dict(
    # _delete_=True,
    batch_size=1,
    num_workers=4,
    persistent_workers=True,
    drop_last=False,
    sampler=dict(type='DefaultSampler', shuffle=False),
    dataset=dict(
        type='ConcatDataset', datasets=test_list, pipeline=test_pipeline))
test_dataloader = val_dataloader

val_evaluator = [
    dict(
        type='MultiDatasetMetricWrapper',
        metric=dict(
            type='WordMetric',
            mode=['exact', 'ignore_case', 'ignore_case_symbol']),
        prefix=['IIIT5K', 'SVT', 'IC13', 'IC15', 'SVTP', 'CUTE80']),    
    dict(
        type='MultiDatasetMetricWrapper',
        metric=dict(
            type='CharMetric',
            mode=['exact', 'ignore_case', 'ignore_case_symbol']),
        prefix=['IIIT5K', 'SVT', 'IC13', 'IC15', 'SVTP', 'CUTE80']),

]
test_evaluator = val_evaluator

Harold-lkk avatar Aug 02 '22 13:08 Harold-lkk