Browse Source

update_seal_det (#2667)

Sunflower7788 11 tháng trước cách đây
mục cha
commit
4a1e27ca86
1 tập tin đã thay đổi với 6 bổ sung3 xóa
  1. 6 3
      paddlex/inference/models_new/text_detection/predictor.py

+ 6 - 3
paddlex/inference/models_new/text_detection/predictor.py

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