diff --git a/deploy/python/preprocess.py b/deploy/python/preprocess.py index aaa0bef3559bbd76d90d6a95389dcc4178d69b2e..5d7fc929675267100f324d014c1e3a4c630b975f 100644 --- a/deploy/python/preprocess.py +++ b/deploy/python/preprocess.py @@ -78,6 +78,9 @@ class UnifiedResize(object): if backend.lower() == "cv2": if isinstance(interpolation, str): interpolation = _cv2_interp_from_str[interpolation.lower()] + # compatible with opencv < version 4.4.0 + elif not interpolation: + interpolation = cv2.INTER_LINEAR self.resize_func = partial(cv2.resize, interpolation=interpolation) elif backend.lower() == "pil": if isinstance(interpolation, str): diff --git a/ppcls/data/preprocess/ops/operators.py b/ppcls/data/preprocess/ops/operators.py index 4418f529356d6dbf1b1c3b33761371b4dbcde3b4..e46823d2a751789d50c98e056fa34b496dee156e 100644 --- a/ppcls/data/preprocess/ops/operators.py +++ b/ppcls/data/preprocess/ops/operators.py @@ -59,6 +59,9 @@ class UnifiedResize(object): if backend.lower() == "cv2": if isinstance(interpolation, str): interpolation = _cv2_interp_from_str[interpolation.lower()] + # compatible with opencv < version 4.4.0 + elif not interpolation: + interpolation = cv2.INTER_LINEAR self.resize_func = partial(cv2.resize, interpolation=interpolation) elif backend.lower() == "pil": if isinstance(interpolation, str):