Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2025-08-04 13:47:37 +02:00
Code Issues Projects Releases Wiki Activity

Move trt error (#19031)

Browse Source 
* Move trt error

* Formatting
This commit is contained in:
Nicolas Mowen 2025-07-07 17:57:37 -06:00 committed by GitHub
parent d69916694b
commit 178117183e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
frigate/detectors/plugins/tensorrt.py
View File

@ -43,13 +43,6 @@ 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):
@ -227,6 +220,14 @@ class TensorRtDetector(DetectionApi):
] ]
def __init__(self, detector_config: TensorRTDetectorConfig): def __init__(self, detector_config: TensorRTDetectorConfig):
if 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."
)
assert TRT_SUPPORT, ( assert TRT_SUPPORT, (
f"TensorRT libraries not found, {DETECTOR_KEY} detector not present" f"TensorRT libraries not found, {DETECTOR_KEY} detector not present"
) )