search

jason-li-831202 / Vehicle-CV-ADAS

The project can achieve FCWS, LDWS, and LKAS functions solely using only visual sensors. using YOLOv5 / YOLOv5-lite / YOLOv6 / YOLOv7 / YOLOv8 / YOLOv9 / EfficientDet and Ultra-Fast-Lane-Detection-v2 .
GNU General Public License v3.0
168 stars 39 forks source link

onxx file convert error ! #27

Open ahmet-f-gumustas opened 2 months ago

ahmet-f-gumustas commented 2 months ago

This is a terminal output :

~/workspace/lane_tracking_detection_with_ai/Vehicle-CV-ADAS$ python3 convertOnnxToTensorRT.py -i /home/moveon2/workspace/lane_tracking_detection_with_ai/Vehicle-CV-ADAS/yolov8l.onnx -o /home/moveon2/workspace/lane_tracking_detection_with_ai/Vehicle-CV-ADAS/model.engine Vehicle-CV-ADAS/convertOnnxToTensorRT: input_onnx_model=/home/moveon2/workspace/lane_tracking_detection_with_ai/Vehicle-CV-ADAS/yolov8l.onnx, output_trt_model=/home/moveon2/workspace/lane_tracking_detection_with_ai/Vehicle-CV-ADAS/model.engine, verbose=False 👉 Starting export with TensorRT Version : 10.0.1 [08/05/2024-11:05:51] [TRT] [I] [MemUsageChange] Init CUDA: CPU +18, GPU +0, now: CPU 24, GPU 750 (MiB) [08/05/2024-11:05:52] [TRT] [I] [MemUsageChange] Init builder kernel library: CPU +1763, GPU +312, now: CPU 1922, GPU 1062 (MiB) Traceback (most recent call last): File "/home/moveon2/workspace/lane_tracking_detection_with_ai/Vehicle-CV-ADAS/convertOnnxToTensorRT.py", line 240, in builder = EngineBuilder(verbose) File "/home/moveon2/workspace/lane_tracking_detection_with_ai/Vehicle-CV-ADAS/convertOnnxToTensorRT.py", line 114, in init self.config.max_workspace_size = int(workspace * (1 << 30)) AttributeError: 'tensorrt_bindings.tensorrt.IBuilderConfig' object has no attribute 'max_workspace_size'

How to solve this error ?

jason-li-831202 commented 2 months ago

Please lower tensorrt version to 8.x.

ahmet-f-gumustas commented 2 months ago

I try soon. Thank You!