From 6c6a87f78e5d8f8c454a2c73910b9fb21dfcb97f Mon Sep 17 00:00:00 2001
From: Tingquan Gao <gaotingquan@baidu.com>
Date: Fri, 28 May 2021 12:20:51 +0800
Subject: [PATCH] Add __init__.py (#750)

* Add __init__.py

* Fix the print info style
---
 paddleclas.py           |  9 +++++++--
 tools/__init__.py       | 15 +++++++++++++++
 tools/infer/__init__.py | 16 ++++++++++++++++
 3 files changed, 38 insertions(+), 2 deletions(-)
 create mode 100644 tools/__init__.py
 create mode 100644 tools/infer/__init__.py

diff --git a/paddleclas.py b/paddleclas.py
index d8fcf936..9d4cdaea 100644
--- a/paddleclas.py
+++ b/paddleclas.py
@@ -160,10 +160,15 @@ class ModelNameError(Exception):
 
 def print_info():
     table = PrettyTable(['Series', 'Name'])
+    try:
+        sz = os.get_terminal_size()
+        width = sz.columns - 30 if sz.columns > 50 else 10
+    except OSError:
+        width = 100
     for series in model_series:
-        names = textwrap.fill("  ".join(model_series[series]), width=100)
+        names = textwrap.fill("  ".join(model_series[series]), width=width)
         table.add_row([series, names])
-    print('Inference models that Paddle provides are listed as follows:')
+    print('Inference models that PaddleClas provides are listed as follows:')
     print(table)
 
 
diff --git a/tools/__init__.py b/tools/__init__.py
new file mode 100644
index 00000000..cdefb805
--- /dev/null
+++ b/tools/__init__.py
@@ -0,0 +1,15 @@
+# Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from . import infer
\ No newline at end of file
diff --git a/tools/infer/__init__.py b/tools/infer/__init__.py
new file mode 100644
index 00000000..8d3600de
--- /dev/null
+++ b/tools/infer/__init__.py
@@ -0,0 +1,16 @@
+# Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from .predict import Predictor
+from .utils import parse_args, create_paddle_predictor, preprocess, postprocess, get_image_list, get_image_list_from_label_file, calc_topk_acc, save_prelabel_results, b64_to_np, np_to_b64
\ No newline at end of file
-- 
GitLab