| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657 |
- '''
- Author: zeke-chin zeke-chin@icloud.com
- Date: 2022-09-06 19:14:40
- LastEditors: zeke-chin zeke-chin@icloud.com
- LastEditTime: 2022-09-22 17:25:24
- FilePath: /to_md/use.py
- Description: 这是默认设置,请设置`customMade`, 打开koroFileHeader查看配置 进行设置: https://github.com/OBKoro1/koro1FileHeader/wiki/%E9%85%8D%E7%BD%AE`
- '''
- from pathlib import Path
- import time
- from new import MD, Image, Dataset, parser_path
- # config
- # 图片路径
- <<<<<<< HEAD
- image_path = Path('/Users/sxkj/Desktop/测试文档/四六级/9.8/身份证/img')
- =======
- image_path = Path('/Users/zeke/work/sx/OCR/image_data/9.15营业执照/img')
- >>>>>>> d3eed34f9749de49c0f16dddb142d9f8befb2890
- image_type = 1
- # 是否旋转
- image_rotate = True
- ocr_address = 'local' # 'local' 'test' 'sb'
- <<<<<<< HEAD
- ocr_name = 'regbook' # 'cet' 'idcard' 'bankcard' 'regbook' 'schoolcert'
- md_name = 'RegBook'
- filed = 'regbook'
- =======
- ocr_name = 'business_license' # 'cet' 'idcard' 'bankcard' 'regbook' 'schoolcert' 'business_license'
- md_name = ocr_name + image_path.name
- filed = ocr_name
- >>>>>>> d3eed34f9749de49c0f16dddb142d9f8befb2890
- # 若md_path为None 则默认使用图片父路径为markdown保存路径
- # md_path = '/Users/zeke/work/sx/OCR/HROCR/to_md/example' or image_path.parent
- md_path = None or image_path.parent
- md_file = parser_path(Path(md_path) / Path(md_name), image_rotate)
- if __name__ == '__main__':
- markdown = MD(md_file)
- dataset = Dataset(image_path, image_type, ocr_name, ocr_address, filed, 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)
- print(md_file)
- markdown.f.create_md_file()
|
