From 79b88c979e7c9d5e9a3f188ce24d26aa75d74aa2 Mon Sep 17 00:00:00 2001 From: Indrek Mandre Date: Fri, 9 Feb 2024 10:45:26 +0200 Subject: [PATCH] detectors/openvino: remove class aggregation --- frigate/detectors/plugins/openvino.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/frigate/detectors/plugins/openvino.py b/frigate/detectors/plugins/openvino.py index fc2ac4682..90b688cbb 100644 --- a/frigate/detectors/plugins/openvino.py +++ b/frigate/detectors/plugins/openvino.py @@ -51,8 +51,6 @@ class OvDetector(DetectionApi): logger.info(f"YOLOX model has {self.num_classes} classes") self.set_strides_grids() - self.class_aggregation = yolo_utils.generate_class_aggregation_from_config(detector_config) - def set_strides_grids(self): grids = [] expanded_strides = [] @@ -138,7 +136,7 @@ class OvDetector(DetectionApi): return detections elif self.ov_model_type == ModelTypeEnum.yolov8: out_tensor = infer_request.get_output_tensor().data - return yolo_utils.yolov8_postprocess(self.interpreter.inputs[0].shape, out_tensor, class_aggregation = self.class_aggregation) + return yolo_utils.yolov8_postprocess(self.interpreter.inputs[0].shape, out_tensor) elif self.ov_model_type == ModelTypeEnum.yolov5: out_tensor = infer_request.get_output_tensor() output_data = out_tensor.data[0]