From c0df38e17a9f3354c635c328ce487f8e87d9bf76 Mon Sep 17 00:00:00 2001 From: Anil Ozyalcin Date: Thu, 16 Feb 2023 20:45:16 -0800 Subject: [PATCH] Fixed double inference bug with YOLOv5 and YOLOv8 --- frigate/detectors/plugins/openvino.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/frigate/detectors/plugins/openvino.py b/frigate/detectors/plugins/openvino.py index 9848232f7..c20e5ebef 100644 --- a/frigate/detectors/plugins/openvino.py +++ b/frigate/detectors/plugins/openvino.py @@ -134,8 +134,6 @@ class OvDetector(DetectionApi): break return detections elif self.ov_model_type == ModelTypeEnum.yolov8: - infer_request = self.interpreter.create_infer_request() - infer_request.infer([tensor_input]) out_tensor = infer_request.get_output_tensor() results = out_tensor.data[0] output_data = np.transpose(results) @@ -170,8 +168,6 @@ class OvDetector(DetectionApi): break return detections elif self.ov_model_type == ModelTypeEnum.yolov5: - infer_request = self.interpreter.create_infer_request() - infer_request.infer([tensor_input]) out_tensor = infer_request.get_output_tensor() output_data = out_tensor.data[0] conf_mask = (output_data[:, 4] >= 0.5).squeeze()