mirror of
https://github.com/blakeblackshear/frigate.git
synced 2026-02-11 13:45:25 +03:00
fix post_process_yolonas call
This commit is contained in:
parent
e4cf53070f
commit
c00a9b255a
@ -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.'
|
||||
|
||||
Loading…
Reference in New Issue
Block a user