administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-03 17:55:21 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Fixed double inference bug with YOLOv5 and YOLOv8

Browse Source
This commit is contained in:
Anil Ozyalcin 2023-02-16 20:45:16 -08:00
parent b561f00ff9
commit c0df38e17a
1 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
frigate/detectors/plugins/openvino.py
View File

@ -134,8 +134,6 @@ class OvDetector(DetectionApi):
break break
return detections return detections
elif self.ov_model_type == ModelTypeEnum.yolov8: 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() out_tensor = infer_request.get_output_tensor()
results = out_tensor.data[0] results = out_tensor.data[0]
output_data = np.transpose(results) output_data = np.transpose(results)
@ -170,8 +168,6 @@ class OvDetector(DetectionApi):
break break
return detections return detections
elif self.ov_model_type == ModelTypeEnum.yolov5: 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() out_tensor = infer_request.get_output_tensor()
output_data = out_tensor.data[0] output_data = out_tensor.data[0]
conf_mask = (output_data[:, 4] >= 0.5).squeeze() conf_mask = (output_data[:, 4] >= 0.5).squeeze()