diff --git a/.gitignore b/.gitignore index 3db08180f2dfb225d3c318b175d6649c123ab0c5..f936b570d1b928eaa8e2ce136b9ce276f3f73876 100644 --- a/.gitignore +++ b/.gitignore @@ -16,3 +16,5 @@ output/ *~ *.vscode *.idea + +*.log diff --git a/README.md b/README.md index 7f0cf32e71e39fa2b48e8b090d8fa25920b23562..0e148f1fa1c9f597ef36e34f2eeda30a69487a8f 100644 --- a/README.md +++ b/README.md @@ -67,13 +67,7 @@ cd .. 以下代码实现了文本检测、识别串联推理,在执行预测时,需要通过参数image_dir指定单张图像或者图像集合的路径、参数det_model_dir指定检测inference模型的路径和参数rec_model_dir指定识别inference模型的路径。可视化识别结果默认保存到 ./inference_results 文件夹里面。 -``` -# 设置PYTHONPATH环境变量 -export PYTHONPATH=. - -# windows下设置环境变量 -SET PYTHONPATH=. - +```bash # 预测image_dir指定的单张图像 python3 tools/infer/predict_system.py --image_dir="./doc/imgs/11.jpg" --det_model_dir="./inference/ch_det_mv3_db/" --rec_model_dir="./inference/ch_rec_mv3_crnn/" diff --git a/README_en.md b/README_en.md index f8aa4e7f753723004d36672a7203557b6e8b5ecb..cfc147c85f6eafb82403f7d0035db27b9fec2f3d 100644 --- a/README_en.md +++ b/README_en.md @@ -67,13 +67,7 @@ cd .. The following code implements text detection and recognition inference tandemly. When performing prediction, you need to specify the path of a single image or image folder through the parameter `image_dir`, the parameter `det_model_dir` specifies the path to detection model, and the parameter `rec_model_dir` specifies the path to the recognition model. The visual prediction results are saved to the `./inference_results` folder by default. -``` -# Set PYTHONPATH environment variable -export PYTHONPATH=. - -# Setting environment variable in Windows -SET PYTHONPATH=. - +```bash # Prediction on a single image by specifying image path to image_dir python3 tools/infer/predict_system.py --image_dir="./doc/imgs/11.jpg" --det_model_dir="./inference/ch_det_mv3_db/" --rec_model_dir="./inference/ch_rec_mv3_crnn/" diff --git a/tools/infer/predict_system.py b/tools/infer/predict_system.py index 3953fa0df76c3929a52af39901f7eeb0569c68b2..1019279a24253d44b7e010adcf286dcd1d58300b 100755 --- a/tools/infer/predict_system.py +++ b/tools/infer/predict_system.py @@ -11,7 +11,11 @@ # 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. - +import os +import sys +__dir__ = os.path.dirname(__file__) +sys.path.append(__dir__) +sys.path.append(os.path.join(__dir__, '../..')) import utility from ppocr.utils.utility import initial_logger logger = initial_logger() @@ -25,7 +29,6 @@ import time from ppocr.utils.utility import get_image_file_list from PIL import Image from tools.infer.utility import draw_ocr -import os class TextSystem(object):