apps/frigate-rockchip
1
0
Fork 0
mirror of https://github.com/MarcA711/frigate-rockchip.git synced 2025-10-08 02:00:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Reimplement support for rknn detector (#11365)

Browse Source 
* initial support for rknn detector

* remove purge_model_cache option

* update rknn

* support rk3576

* fix post_process_yolonas call

* add yolonas models

* update config

* exclude yolonas from image

* remove code
This commit is contained in:
Marc Altmann
2024-05-22 00:50:03 +02:00
committed by GitHub
parent 910c85b1c0
commit e91f3d8d9b
7 changed files with 231 additions and 90 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
frigate/detectors/detection_api.py
View File

@@ -72,7 +72,7 @@ class DetectionApi(ABC):
def post_process(self, output):
if self.detector_config.model.model_type == ModelTypeEnum.yolonas:
return self.yolonas(output)
return self.post_process_yolonas(output)
else:
raise ValueError(
f'Model type "{self.detector_config.model.model_type}" is currently not supported.'