"git@developer.sourcefind.cn:OpenDAS/apex.git" did not exist on "6e453f1a254929f5c035d73fc4b050284ce0ce38"
Commit f8323ae0 authored by myhloli's avatar myhloli
Browse files

refactor(ocr): comment out det_count update and update OCR models

- Comment out the line that updates det_count in batch_analyze.py
- Add a new OCR model configuration for Chinese (ch_lite) in models_config.yml- Update the Chinese OCR model configuration to use a different recognition model
parent 814bd4ea
...@@ -143,7 +143,8 @@ class BatchAnalyze: ...@@ -143,7 +143,8 @@ class BatchAnalyze:
if ocr_res: if ocr_res:
ocr_result_list = get_ocr_result_list(ocr_res, useful_list, ocr_res_list_dict['ocr_enable'], new_image, _lang) ocr_result_list = get_ocr_result_list(ocr_res, useful_list, ocr_res_list_dict['ocr_enable'], new_image, _lang)
ocr_res_list_dict['layout_res'].extend(ocr_result_list) ocr_res_list_dict['layout_res'].extend(ocr_result_list)
det_count += len(ocr_res_list_dict['ocr_res_list'])
# det_count += len(ocr_res_list_dict['ocr_res_list'])
# logger.info(f'ocr-det time: {round(time.time()-det_start, 2)}, image num: {det_count}') # logger.info(f'ocr-det time: {round(time.time()-det_start, 2)}, image num: {det_count}')
......
lang: lang:
ch: ch_lite:
det: ch_PP-OCRv3_det_infer.pth det: ch_PP-OCRv3_det_infer.pth
rec: ch_PP-OCRv4_rec_infer.pth rec: ch_PP-OCRv4_rec_infer.pth
dict: ppocr_keys_v1.txt dict: ppocr_keys_v1.txt
ch:
det: ch_PP-OCRv3_det_infer.pth
rec: ch_PP-OCRv4_rec_server_infer.pth
dict: ppocr_keys_v1.txt
en: en:
det: en_PP-OCRv3_det_infer.pth det: en_PP-OCRv3_det_infer.pth
rec: en_PP-OCRv4_rec_infer.pth rec: en_PP-OCRv4_rec_infer.pth
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment