docker file

bak.Dockerfile → Dockerfile

@@ -21,7 +21,7 @@ USER user
 ENV HOME=/home/user
 WORKDIR $HOME/app
 COPY --chown=user . $HOME/app
-RUN paddleocr --image_dir ./example_imgs/word_1.jpg --use_angle_cls true --use_gpu false  --lang ch
-RUN paddleocr --image_dir ./example_imgs/example.jpg --use_angle_cls true --use_gpu false  --lang en
+# RUN paddleocr --image_dir ./example_imgs/word_1.jpg --use_angle_cls true --use_gpu false  --lang ch --det_model_dir ./models/det/ch --cls_model_dir ./models/cls/ch --rec_model_dir ./models/rec/ch
+# RUN paddleocr --image_dir ./example_imgs/example.jpg --use_angle_cls true --use_gpu false  --lang en --det_model_dir ./models/det/en --cls_model_dir ./models/cls/en --rec_model_dir ./models/rec/en
 
 CMD ["python", "-u", "app.py"]

app.py

@@ -22,7 +22,10 @@ def draw_ocr_bbox(image, boxes, colors):
 # torch.hub.download_url_to_file('https://i.imgur.com/aqMBT0i.jpg', 'example.jpg')
 
 def inference(img: Image.Image, lang, confidence):
-    ocr = PaddleOCR(use_angle_cls=True, lang=lang, use_gpu=False)
+    ocr = PaddleOCR(use_angle_cls=True, lang=lang, use_gpu=False,
+                    det_model_dir=f'./models/det/{lang}',
+                    cls_model_dir=f'./models/cls/{lang}',
+                    rec_model_dir=f'./models/rec/{lang}')
     # img_path = img.name
     img2np = np.array(img)
     result = ocr.ocr(img2np, cls=True)[0]