Merge pull request #8020 from WenmuZhou/whl_2.6

add recovery requirements to whl

paddleocr.py

@@ -47,7 +47,7 @@ __all__ = [
 ]
 
 SUPPORT_DET_MODEL = ['DB']
-VERSION = '2.6.0.1'
+VERSION = '2.6.0.3'
 SUPPORT_REC_MODEL = ['CRNN', 'SVTR_LCNet']
 BASE_DIR = os.path.expanduser("~/.paddleocr/")
 

ppstructure/docs/quickstart.md

@@ -49,12 +49,6 @@ pip3 install "paddleocr>=2.6"
 
 # 安装 图像方向分类依赖包paddleclas（如不需要图像方向分类功能，可跳过）
 pip3 install paddleclas>=2.4.3
-
-# 安装 关键信息抽取 依赖包（如不需要KIE功能，可跳过）
-pip3 install -r ppstructure/kie/requirements.txt
-
-# 安装 版面恢复 依赖包（如不需要版面恢复功能，可跳过）
-pip3 install -r ppstructure/recovery/requirements.txt
 ```
 
 <a name="2"></a>

ppstructure/docs/quickstart_en.md

@@ -51,12 +51,6 @@ pip3 install "paddleocr>=2.6"
 
 # Install the image direction classification dependency package paddleclas (if you do not use the image direction classification, you can skip it)
 pip3 install paddleclas>=2.4.3
-
-# Install the KIE dependency packages (if you do not use the KIE, you can skip it)
-pip3 install -r kie/requirements.txt
-
-# Install the layout recovery dependency packages (if you do not use the layout recovery, you can skip it)
-pip3 install -r recovery/requirements.txt
 ```
 
 <a name="2"></a>

setup.py

@@ -16,9 +16,17 @@ from setuptools import setup
 from io import open
 from paddleocr import VERSION
 
-with open('requirements.txt', encoding="utf-8-sig") as f:
-    requirements = f.readlines()
-    requirements.append('tqdm')
+
+def load_requirements(file_list=None):
+    if file_list is None:
+        file_list = ['requirements.txt']
+    if isinstance(file_list, str):
+        file_list = [file_list]
+    requirements = []
+    for file in file_list:
+        with open(file, encoding="utf-8-sig") as f:
+            requirements.extend(f.readlines())
+    return requirements
 
 
 def readme():
@@ -34,7 +42,8 @@ setup(
     include_package_data=True,
     entry_points={"console_scripts": ["paddleocr= paddleocr.paddleocr:main"]},
     version=VERSION,
-    install_requires=requirements,
+    install_requires=load_requirements(
+        ['requirements.txt', 'ppstructure/recovery/requirements.txt']),
     license='Apache License 2.0',
     description='Awesome OCR toolkits based on PaddlePaddle （8.6M ultra-lightweight pre-trained model, support training and deployment among server, mobile, embeded and IoT devices',
     long_description=readme(),