Ver código fonte

update_seal_det (#2667)

Sunflower7788 11 meses atrás
pai
commit
4a1e27ca86

+ 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(