Skip to content

Commit 4a1e27c

Browse files
authored
update_seal_det (#2667)
1 parent 36c0a29 commit 4a1e27c

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

paddlex/inference/models_new/text_detection/predictor.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -90,9 +90,12 @@ def build_readimg(self, channel_first, img_mode):
9090
@register("DetResizeForTest")
9191
def build_resize(self, **kwargs):
9292
# TODO: align to PaddleOCR
93-
assert self.model_name in ("PP-OCRv4_server_det", "PP-OCRv4_mobile_det")
94-
resize_long = kwargs.get("resize_long", 960)
95-
return "Resize", DetResizeForTest(limit_side_len=resize_long, limit_type="max")
93+
if self.model_name in ("PP-OCRv4_server_det", "PP-OCRv4_mobile_det"):
94+
resize_long = kwargs.get("resize_long", 960)
95+
return "Resize", DetResizeForTest(
96+
limit_side_len=resize_long, limit_type="max"
97+
)
98+
return "Resize", DetResizeForTest(**kwargs)
9699

97100
@register("NormalizeImage")
98101
def build_normalize(

0 commit comments

Comments
 (0)