Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Trt and onnx files source #1

Open
hamaadtahiir opened this issue Feb 15, 2023 · 8 comments
Open

Trt and onnx files source #1

hamaadtahiir opened this issue Feb 15, 2023 · 8 comments
Labels
documentation Improvements or additions to documentation

Comments

@hamaadtahiir
Copy link

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

@jason-li-831202
Copy link
Owner

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".

@nqkhanh2002
Copy link

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

@jason-li-831202
Copy link
Owner

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?

@nqkhanh2002
Copy link

@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

@jason-li-831202 jason-li-831202 added the documentation Improvements or additions to documentation label Jun 7, 2023
@balajik27
Copy link

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
Copy link

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

@Jilin-Cancer
Copy link

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!!!

1 similar comment
@Sadcato
Copy link

Sadcato commented Feb 21, 2024

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!!!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
documentation Improvements or additions to documentation
Projects
None yet
Development

No branches or pull requests

6 participants