From 6b80061cfd2bf6346bbf9cb9c76194c730d694cb Mon Sep 17 00:00:00 2001 From: knoffelcut Date: Sat, 30 May 2026 18:41:08 +0200 Subject: [PATCH] NanoDet-Plus OpenVINO compatibility --- frigate/detectors/plugins/openvino.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/frigate/detectors/plugins/openvino.py b/frigate/detectors/plugins/openvino.py index 1e9fb1ab10..2be8e527cc 100644 --- a/frigate/detectors/plugins/openvino.py +++ b/frigate/detectors/plugins/openvino.py @@ -10,6 +10,7 @@ from frigate.detectors.detection_runners import OpenVINOModelRunner from frigate.detectors.detector_config import BaseDetectorConfig, ModelTypeEnum from frigate.util.model import ( post_process_dfine, + post_process_nanodet_plus, post_process_rfdetr, post_process_yolo, ) @@ -43,6 +44,7 @@ class OvDetector(DetectionApi): ModelTypeEnum.yolonas, ModelTypeEnum.yologeneric, ModelTypeEnum.yolox, + ModelTypeEnum.nanodet_plus, ] def __init__(self, detector_config: OvDetectorConfig): @@ -238,3 +240,9 @@ class OvDetector(DetectionApi): object_detected[6], object_detected[5], object_detected[:4] ) return detections + elif self.ov_model_type == ModelTypeEnum.nanodet_plus: + return post_process_nanodet_plus( + outputs[0], + self.width, + self.height, + )