瀏覽代碼

fixed ocrV5 det model bug (#4039)

学卿 6 月之前
父節點
當前提交
a53888130d
共有 1 個文件被更改,包括 2 次插入0 次删除
  1. 2 0
      paddlex/inference/models/text_detection/predictor.py

+ 2 - 0
paddlex/inference/models/text_detection/predictor.py

@@ -133,6 +133,8 @@ class TextDetPredictor(BasePredictor):
         # TODO: align to PaddleOCR
         # TODO: align to PaddleOCR
 
 
         if self.model_name in (
         if self.model_name in (
+            "PP-OCRv5_server_det",
+            "PP-OCRv5_mobile_det",
             "PP-OCRv4_server_det",
             "PP-OCRv4_server_det",
             "PP-OCRv4_mobile_det",
             "PP-OCRv4_mobile_det",
             "PP-OCRv3_server_det",
             "PP-OCRv3_server_det",