Vehicle-CV-ADAS icon indicating copy to clipboard operation
Vehicle-CV-ADAS copied to clipboard

Published 20 hours ago • verified publisher icon jason-li-831202

Trt and onnx files source

Open hamaadtahiir opened this issue 2 years ago • 8 comments

From where can we get the onnx and/or trt files for the models?

hamaadtahiir avatar Feb 15 '23 17:02 hamaadtahiir

you can download YOLOv8 onnx model from https://github.com/ibaiGorordo/ONNX-YOLOv8-Object-Detection image

Ultra-Fast-Lane-Detection-V2 pt nodel from https://github.com/cfzd/Ultra-Fast-Lane-Detection-v2 image , and use convertPytorchToONNX.py in TrafficLaneDetector Folder convet to onnx model.

At last, all onnx model can convert to trt just use convertOnnxToTensorRT.py , remember to modify "onnx_model_path" and "trt_model_path".

jason-li-831202 avatar Feb 16 '23 03:02 jason-li-831202

Can you provide a link to the drive or zip file or similar, I'm stuck in converting for hours, thanks for that.

nqkhanh2002 avatar Jun 04 '23 00:06 nqkhanh2002

I can provide a temporary link : https://drive.google.com/drive/folders/1lo_nFwvrm9ey0dj6FbyAaBxCzowol77_?usp=sharing

Perhaps you can share some difficulties you encountered during the converting?

jason-li-831202 avatar Jun 04 '23 02:06 jason-li-831202

@jason-li-831202 I'm having a hard time working with the NIVIDA GPU so I'm putting the code on colab and it looks good. By the way can I get the test file for you, I have tried searching the internet like no accident videos and the road conditions are pretty good. Thank you very much

nqkhanh2002 avatar Jun 04 '23 08:06 nqkhanh2002

you can download YOLOv8 onnx model from https://github.com/ibaiGorordo/ONNX-YOLOv8-Object-Detection image

Ultra-Fast-Lane-Detection-V2 pt nodel from https://github.com/cfzd/Ultra-Fast-Lane-Detection-v2 image , and use convertPytorchToONNX.py in TrafficLaneDetector Folder convet to onnx model.

At last, all onnx model can convert to trt just use convertOnnxToTensorRT.py , remember to modify "onnx_model_path" and "trt_model_path".

where i can get yolov8m-coco.onnx file, from the link u have given i didnt get that file.

balajik27 avatar Sep 24 '23 16:09 balajik27

I can provide a temporary link : https://drive.google.com/drive/folders/1lo_nFwvrm9ey0dj6FbyAaBxCzowol77_?usp=sharing

Perhaps you can share some difficulties you encountered during the converting?

this link shows 404 error

balajik27 avatar Sep 24 '23 16:09 balajik27

I can provide a temporary link : https://drive.google.com/drive/folders/1lo_nFwvrm9ey0dj6FbyAaBxCzowol77_?usp=sharing

Perhaps you can share some difficulties you encountered during the converting?

Hello master, your work is great, I am very interested, your link is invalid, can you share it again, thank you!!!

Jilin-Cancer avatar Nov 28 '23 09:11 Jilin-Cancer

I can provide a temporary link : https://drive.google.com/drive/folders/1lo_nFwvrm9ey0dj6FbyAaBxCzowol77_?usp=sharing

Perhaps you can share some difficulties you encountered during the converting?

Hello master, your work is great, I am very interested, your link is invalid, can you share it again, thank you!!!

Sadcato avatar Feb 21 '24 05:02 Sadcato