ソースを参照

fix params of resize op in ocr_det (#2246)

Tingquan Gao 1 年間 前
コミット
cf46388436
1 ファイル変更6 行追加2 行削除
  1. 6 2
      paddlex/inference/models/text_detection.py

+ 6 - 2
paddlex/inference/models/text_detection.py

@@ -53,8 +53,12 @@ class TextDetPredictor(BasicPredictor):
         return ReadImage(format=img_mode)
 
     @register("DetResizeForTest")
-    def build_resize(self, resize_long=960):
-        return DetResizeForTest(limit_side_len=resize_long, limit_type="max")
+    def build_resize(self, **kwargs):
+        # TODO: align to PaddleOCR
+        if self.model_name in ("PP-OCRv4_server_det", "PP-OCRv4_mobile_det"):
+            resize_long = kwargs.get("resize_long", 960)
+            return DetResizeForTest(limit_side_len=resize_long, limit_type="max")
+        return DetResizeForTest(**kwargs)
 
     @register("NormalizeImage")
     def build_normalize(