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

Add YOLOv9 support to RKNN (#17791)

Browse Source 
* Add yolov9

* Undo

* Update docs for rknn yolov9

* Update docs notes

* Add infernece times table
This commit is contained in:
Nicolas Mowen
2025-04-18 16:51:04 -06:00
committed by GitHub
parent e8883a2a2e
commit 89b54f19c8
3 changed files with 32 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
frigate/detectors/plugins/rknn.py
+3
View File
@@ -11,6 +11,7 @@ from pydantic import Field
from frigate.const import MODEL_CACHE_DIR
from frigate.detectors.detection_api import DetectionApi
from frigate.detectors.detector_config import BaseDetectorConfig, ModelTypeEnum
from frigate.util.model import post_process_yolo
logger = logging.getLogger(__name__)
@@ -284,6 +285,8 @@ class Rknn(DetectionApi):
def post_process(self, output):
if self.detector_config.model.model_type == ModelTypeEnum.yolonas:
return self.post_process_yolonas(output)
elif self.detector_config.model.model_type == ModelTypeEnum.yologeneric:
return post_process_yolo(output, self.width, self.height)
elif self.detector_config.model.model_type == ModelTypeEnum.yolox:
return self.post_process_yolox(output, self.grids, self.expanded_strides)
else: