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

detectors/openvino: remove class aggregation

Browse Source
This commit is contained in:
Indrek Mandre 2024-02-09 10:45:26 +02:00
parent 4dc7502f8a
commit 79b88c979e
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -51,8 +51,6 @@ class OvDetector(DetectionApi):
logger.info(f"YOLOX model has {self.num_classes} classes") logger.info(f"YOLOX model has {self.num_classes} classes")
self.set_strides_grids() self.set_strides_grids()
self.class_aggregation = yolo_utils.generate_class_aggregation_from_config(detector_config)
def set_strides_grids(self): def set_strides_grids(self):
grids = [] grids = []
expanded_strides = [] expanded_strides = []
@ -138,7 +136,7 @@ class OvDetector(DetectionApi):
return detections return detections
elif self.ov_model_type == ModelTypeEnum.yolov8: elif self.ov_model_type == ModelTypeEnum.yolov8:
out_tensor = infer_request.get_output_tensor().data 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: elif self.ov_model_type == ModelTypeEnum.yolov5:
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]