Add logs for directing users not to use tensorrt detector

This commit is contained in:
Nicolas Mowen 2025-07-07 11:18:09 -06:00
parent 9cccef6444
commit 07b5f67cc1

View File

@ -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):