fix: interpolation maybe be 0

0dccfb91 · gaotingquan · 1c01d7a7 · 0dccfb91 · 0dccfb91
显示空白变更内容
内联并排

Showing with 2 addition and 2 deletion

deploy/python/preprocess.py deploy/python/preprocess.py +1 -1

ppcls/data/preprocess/ops/operators.py ppcls/data/preprocess/ops/operators.py +1 -1

未找到文件。
--- a/deploy/python/preprocess.py
+++ b/deploy/python/preprocess.py
@@ -79,7 +79,7 @@ class UnifiedResize(object):
            if isinstance(interpolation, str):
                interpolation = _cv2_interp_from_str[interpolation.lower()]
            # compatible with opencv < version 4.4.0
-            elif not interpolation:
+            elif interpolation is None:
                interpolation = cv2.INTER_LINEAR
            self.resize_func = partial(cv2.resize, interpolation=interpolation)
        elif backend.lower() == "pil":

--- a/ppcls/data/preprocess/ops/operators.py
+++ b/ppcls/data/preprocess/ops/operators.py
@@ -60,7 +60,7 @@ class UnifiedResize(object):
            if isinstance(interpolation, str):
                interpolation = _cv2_interp_from_str[interpolation.lower()]
            # compatible with opencv < version 4.4.0
-            elif not interpolation:
+            elif interpolation is None:
                interpolation = cv2.INTER_LINEAR
            self.resize_func = partial(cv2.resize, interpolation=interpolation)
        elif backend.lower() == "pil":