hr-ocr-cet/markdown/use_new.py

from pathlib import Path

from markdown.new import MD, Image, Dataset

# config
image_path = Path('images/test/image')
image_type = None
image_rotate = True
ocr_address = 'local'  # 'local' 'test' 'sb'
ocr_name = 'cet'  # 'cet' 'idcard' 'bankcard' 'regbook' 'schoolcert'

if __name__ == '__main__':
    markdown = MD('英语等级证书')

    dataset = Dataset(image_path, image_type, ocr_name, ocr_address, image_rotate)
    print(len(dataset))
    for d in dataset():
        print(d)

    dataset.evaluate()
    print(dataset.accuracy)

    markdown.write_total_accuracy(dataset)
    markdown.write_table_accuracy(dataset)
    markdown.write_table_result(dataset)

    markdown.f.create_md_file()