mirror of
https://github.com/blakeblackshear/frigate.git
synced 2026-04-27 17:17:40 +03:00
Add logs for directing users not to use tensorrt detector
This commit is contained in:
parent
9cccef6444
commit
07b5f67cc1
@ -1,5 +1,6 @@
|
|||||||
import ctypes
|
import ctypes
|
||||||
import logging
|
import logging
|
||||||
|
import platform
|
||||||
|
|
||||||
import numpy as np
|
import numpy as np
|
||||||
|
|
||||||
@ -42,6 +43,13 @@ if TRT_SUPPORT:
|
|||||||
return logging.CRITICAL
|
return logging.CRITICAL
|
||||||
else:
|
else:
|
||||||
return logging.DEBUG
|
return logging.DEBUG
|
||||||
|
elif platform.machine() == "x86_64":
|
||||||
|
logger.error(
|
||||||
|
"TensorRT detector is no longer supported on amd64 system. Please use ONNX detector instead, see https://docs.frigate.video/configuration/object_detectors#onnx for more information."
|
||||||
|
)
|
||||||
|
raise ImportError(
|
||||||
|
"TensorRT detector is no longer supported on amd64 system. Please use ONNX detector instead, see https://docs.frigate.video/configuration/object_detectors#onnx for more information."
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
class TensorRTDetectorConfig(BaseDetectorConfig):
|
class TensorRTDetectorConfig(BaseDetectorConfig):
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user