mirror of
https://github.com/blakeblackshear/frigate.git
synced 2026-04-24 15:47:43 +03:00
remove unnecessary hailo resize
This commit is contained in:
parent
4efa37808f
commit
b9001ae2f0
@ -33,10 +33,6 @@ def preprocess_tensor(image: np.ndarray, model_w: int, model_h: int) -> np.ndarr
|
|||||||
image = image[0]
|
image = image[0]
|
||||||
|
|
||||||
h, w = image.shape[:2]
|
h, w = image.shape[:2]
|
||||||
|
|
||||||
if (w, h) == (320, 320) and (model_w, model_h) == (640, 640):
|
|
||||||
return cv2.resize(image, (model_w, model_h), interpolation=cv2.INTER_LINEAR)
|
|
||||||
|
|
||||||
scale = min(model_w / w, model_h / h)
|
scale = min(model_w / w, model_h / h)
|
||||||
new_w, new_h = int(w * scale), int(h * scale)
|
new_w, new_h = int(w * scale), int(h * scale)
|
||||||
resized_image = cv2.resize(image, (new_w, new_h), interpolation=cv2.INTER_CUBIC)
|
resized_image = cv2.resize(image, (new_w, new_h), interpolation=cv2.INTER_CUBIC)
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user