administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-05-05 13:07:44 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Show correct error when wrong model is used

Browse Source
This commit is contained in:
Nicolas Mowen 2025-03-14 16:59:01 -06:00
parent 19342c8768
commit 1a0e0f7eb3
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
frigate/detectors/plugins/edgetpu_tfl.py
View File

@ -1,4 +1,5 @@
import logging
import os
import numpy as np
from pydantic import Field
@ -45,9 +46,15 @@ class EdgeTpuTfl(DetectionApi):
experimental_delegates=[edge_tpu_delegate],
)
except ValueError:
logger.error(
"No EdgeTPU was detected. If you do not have a Coral device yet, you must configure CPU detectors."
)
_, ext = os.path.splitext(detector_config.model.path)
if ext and ext != ".tflite":
logger.error("Incorrect model used with EdgeTPU. Only .tflite models can be used with a Coral EdgeTPU.")
else:
logger.error(
"No EdgeTPU was detected. If you do not have a Coral device yet, you must configure CPU detectors."
)
raise
self.interpreter.allocate_tensors()