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

Add default model for ArmNN - it shares with CPU

Browse Source
This commit is contained in:
RichardPar 2023-05-13 21:59:34 +01:00
parent 9394e57b31
commit 1e78e48ec8
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
frigate/config.py
View File

@ -1062,6 +1062,9 @@ class FrigateConfig(FrigateBaseModel):
merged_model["path"] = "/cpu_model.tflite" merged_model["path"] = "/cpu_model.tflite"
elif detector_config.type == "edgetpu": elif detector_config.type == "edgetpu":
merged_model["path"] = "/edgetpu_model.tflite" merged_model["path"] = "/edgetpu_model.tflite"
elif detector_config.type == "armgpu":
+ merged_model["path"] = "/cpu_model.tflite"
detector_config.model = ModelConfig.parse_obj(merged_model) detector_config.model = ModelConfig.parse_obj(merged_model)
detector_config.model.check_and_load_plus_model( detector_config.model.check_and_load_plus_model(