From add5e2a256f6143f765ffaf834615baff8a6e15b Mon Sep 17 00:00:00 2001 From: WenmuZhou Date: Tue, 8 Jun 2021 18:24:07 +0800 Subject: [PATCH] Make drawing functions available externally --- ppstructure/__init__.py | 4 ++-- ppstructure/paddlestructure.py | 2 +- ppstructure/setup.py | 2 +- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/ppstructure/__init__.py b/ppstructure/__init__.py index e4fc8196..7055bee4 100644 --- a/ppstructure/__init__.py +++ b/ppstructure/__init__.py @@ -12,6 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -from .paddlestructure import PaddleStructure +from .paddlestructure import PaddleStructure, draw_result, to_excel -__all__ = ['PaddleStructure'] \ No newline at end of file +__all__ = ['PaddleStructure', 'draw_result', 'to_excel'] diff --git a/ppstructure/paddlestructure.py b/ppstructure/paddlestructure.py index 0cf5e9d0..57a53d64 100644 --- a/ppstructure/paddlestructure.py +++ b/ppstructure/paddlestructure.py @@ -145,4 +145,4 @@ def main(): for item in result: logger.info(item['res']) save_res(result, save_folder, img_name) - logger.info('result save to {}'.format(os.path.join(save_folder, img_name))) + logger.info('result save to {}'.format(os.path.join(save_folder, img_name))) \ No newline at end of file diff --git a/ppstructure/setup.py b/ppstructure/setup.py index 0d7b2b9a..8e68b2e4 100644 --- a/ppstructure/setup.py +++ b/ppstructure/setup.py @@ -43,7 +43,7 @@ setup( package_dir={'paddlestructure': ''}, include_package_data=True, entry_points={"console_scripts": ["paddlestructure= paddlestructure.paddlestructure:main"]}, - version='2.0.6', + version='1.0', install_requires=requirements, 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', -- GitLab