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

fix post_process_yolonas call

Browse Source
This commit is contained in:
MarcA711 2024-05-20 09:26:51 +00:00
parent e4cf53070f
commit c00a9b255a
1 changed files with 1 additions and 1 deletions
Show Stats 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): def post_process(self, output):
if self.detector_config.model.model_type == ModelTypeEnum.yolonas: if self.detector_config.model.model_type == ModelTypeEnum.yolonas:
return self.yolonas(output) return self.post_process_yolonas(output)
else: else:
raise ValueError( raise ValueError(
f'Model type "{self.detector_config.model.model_type}" is currently not supported.' f'Model type "{self.detector_config.model.model_type}" is currently not supported.'