"...git@developer.sourcefind.cn:chenpangpang/open-webui.git" did not exist on "0b62bbb52eab853e20ee85a2a855518f2bc2cbc7"
Commit f85b3ea8 authored by WenmuZhou's avatar WenmuZhou
Browse files

fix bug in whl import fce

parent 496887f8
include LICENSE include LICENSE
include README.md include README.md
recursive-include ppocr/utils *.txt utility.py logging.py network.py recursive-include ppocr/utils *.*
recursive-include ppocr/data *.py recursive-include ppocr/data *.py
recursive-include ppocr/postprocess *.py recursive-include ppocr/postprocess *.py
recursive-include tools/infer *.py recursive-include tools/infer *.py
......
...@@ -42,7 +42,7 @@ __all__ = [ ...@@ -42,7 +42,7 @@ __all__ = [
] ]
SUPPORT_DET_MODEL = ['DB'] SUPPORT_DET_MODEL = ['DB']
VERSION = '2.4.0.1' VERSION = '2.4.0.2'
SUPPORT_REC_MODEL = ['CRNN'] SUPPORT_REC_MODEL = ['CRNN']
BASE_DIR = os.path.expanduser("~/.paddleocr/") BASE_DIR = os.path.expanduser("~/.paddleocr/")
...@@ -308,20 +308,18 @@ class PaddleOCR(predict_system.TextSystem): ...@@ -308,20 +308,18 @@ class PaddleOCR(predict_system.TextSystem):
det_lang) det_lang)
params.det_model_dir, det_url = confirm_model_dir_url( params.det_model_dir, det_url = confirm_model_dir_url(
params.det_model_dir, params.det_model_dir,
os.path.join(BASE_DIR, VERSION, 'ocr', 'det', det_lang), os.path.join(BASE_DIR, 'whl', 'det', det_lang),
det_model_config['url']) det_model_config['url'])
rec_model_config = get_model_config('OCR', params.ocr_version, 'rec', rec_model_config = get_model_config('OCR', params.ocr_version, 'rec',
lang) lang)
params.rec_model_dir, rec_url = confirm_model_dir_url( params.rec_model_dir, rec_url = confirm_model_dir_url(
params.rec_model_dir, params.rec_model_dir,
os.path.join(BASE_DIR, VERSION, 'ocr', 'rec', lang), os.path.join(BASE_DIR, 'whl', 'rec', lang), rec_model_config['url'])
rec_model_config['url'])
cls_model_config = get_model_config('OCR', params.ocr_version, 'cls', cls_model_config = get_model_config('OCR', params.ocr_version, 'cls',
'ch') 'ch')
params.cls_model_dir, cls_url = confirm_model_dir_url( params.cls_model_dir, cls_url = confirm_model_dir_url(
params.cls_model_dir, params.cls_model_dir,
os.path.join(BASE_DIR, VERSION, 'ocr', 'cls'), os.path.join(BASE_DIR, 'whl', 'cls'), cls_model_config['url'])
cls_model_config['url'])
# download model # download model
maybe_download(params.det_model_dir, det_url) maybe_download(params.det_model_dir, det_url)
maybe_download(params.rec_model_dir, rec_url) maybe_download(params.rec_model_dir, rec_url)
...@@ -412,20 +410,18 @@ class PPStructure(OCRSystem): ...@@ -412,20 +410,18 @@ class PPStructure(OCRSystem):
det_lang) det_lang)
params.det_model_dir, det_url = confirm_model_dir_url( params.det_model_dir, det_url = confirm_model_dir_url(
params.det_model_dir, params.det_model_dir,
os.path.join(BASE_DIR, VERSION, 'ocr', 'det', det_lang), os.path.join(BASE_DIR, 'whl', 'det', det_lang),
det_model_config['url']) det_model_config['url'])
rec_model_config = get_model_config('OCR', params.ocr_version, 'rec', rec_model_config = get_model_config('OCR', params.ocr_version, 'rec',
lang) lang)
params.rec_model_dir, rec_url = confirm_model_dir_url( params.rec_model_dir, rec_url = confirm_model_dir_url(
params.rec_model_dir, params.rec_model_dir,
os.path.join(BASE_DIR, VERSION, 'ocr', 'rec', lang), os.path.join(BASE_DIR, 'whl', 'rec', lang), rec_model_config['url'])
rec_model_config['url'])
table_model_config = get_model_config( table_model_config = get_model_config(
'STRUCTURE', params.structure_version, 'table', 'en') 'STRUCTURE', params.structure_version, 'table', 'en')
params.table_model_dir, table_url = confirm_model_dir_url( params.table_model_dir, table_url = confirm_model_dir_url(
params.table_model_dir, params.table_model_dir,
os.path.join(BASE_DIR, VERSION, 'ocr', 'table'), os.path.join(BASE_DIR, 'whl', 'table'), table_model_config['url'])
table_model_config['url'])
# download model # download model
maybe_download(params.det_model_dir, det_url) maybe_download(params.det_model_dir, det_url)
maybe_download(params.rec_model_dir, rec_url) maybe_download(params.rec_model_dir, rec_url)
...@@ -499,4 +495,4 @@ def main(): ...@@ -499,4 +495,4 @@ def main():
for item in result: for item in result:
item.pop('img') item.pop('img')
logger.info(item) logger.info(item)
\ No newline at end of file
...@@ -622,7 +622,6 @@ def get_rotate_crop_image(img, points): ...@@ -622,7 +622,6 @@ def get_rotate_crop_image(img, points):
def check_gpu(use_gpu): def check_gpu(use_gpu):
if use_gpu and not paddle.is_compiled_with_cuda(): if use_gpu and not paddle.is_compiled_with_cuda():
use_gpu = False use_gpu = False
return use_gpu return use_gpu
......
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