From cc807f49a045c59926a914a8fb29381bcd8fc07e Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Thu, 17 Apr 2025 17:09:40 -0600 Subject: [PATCH] Handle non normalized float --- frigate/object_detection/base.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/frigate/object_detection/base.py b/frigate/object_detection/base.py index dfc39ac2d..b837984fa 100644 --- a/frigate/object_detection/base.py +++ b/frigate/object_detection/base.py @@ -77,6 +77,8 @@ class LocalObjectDetector(ObjectDetector): if self.dtype == InputDTypeEnum.float: tensor_input = tensor_input.astype(np.float32) tensor_input /= 255 + elif self.dtype == InputDTypeEnum.float_denorm: + tensor_input = tensor_input.astype(np.float32) return self.detect_api.detect_raw(tensor_input=tensor_input)