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

Can I run on RTX 4070 ? #197

Open
mitesh741 opened this issue Oct 7, 2023 · 8 comments
Open

Can I run on RTX 4070 ? #197

mitesh741 opened this issue Oct 7, 2023 · 8 comments

Comments

@mitesh741
Copy link

Hi I want to run this repo on my RTX 4070. I do not have ZED Camera. I do not want to use it. Is there any way to run YoloP model only with Tensorrt?

Much Appreciated.

Thanks in Advance.

@HunterShinobiTitan
Copy link

Tested on my RTX 3050

@HunterShinobiTitan
Copy link

you can use the repo from https://github.com/wang-xinyu/tensorrtx/tree/master/yolop

@gwy0o0
Copy link

gwy0o0 commented Nov 23, 2023

I would like to ask how to show the confidence and category of vehicle detection in YOLOP test results, hoping to get your answer.

@HunterShinobiTitan
Copy link

I would like to ask how to show the confidence and category of vehicle detection in YOLOP test results, hoping to get your answer.
just uncomment the code inside plot_one_box() from plot.py

@gwy0o0
Copy link

gwy0o0 commented Nov 26, 2023

I would like to ask how to show the confidence and category of vehicle detection in YOLOP test results, hoping to get your answer.
just uncomment the code inside plot_one_box() from plot.py

This method is very effective. But the displayed category is 0. Is there any way to display the category name, such as car? In addition, I would like to ask if YOLOP retraining for multi-target detection is not effective and is there any way to improve this effect? Is it feasible to add another detector? Thank you for your answer!

@HunterShinobiTitan
Copy link

HunterShinobiTitan commented Nov 26, 2023

I would like to ask how to show the confidence and category of vehicle detection in YOLOP test results, hoping to get your answer.
just uncomment the code inside plot_one_box() from plot.py

This method is very effective. But the displayed category is 0. Is there any way to display the category name, such as car? In addition, I would like to ask if YOLOP retraining for multi-target detection is not effective and is there any way to improve this effect? Is it feasible to add another detector? Thank you for your answer!

You can use the dictionary from convert.py you can use and given category with different keys.

if you are using demo.py you can modify the code for the label prediction.

@HunterShinobiTitan
Copy link

I suggest to use custom dataset, if you want to improve multitarget detection.

@gwy0o0
Copy link

gwy0o0 commented Nov 26, 2023

I suggest to use custom dataset, if you want to improve multitarget detection.

Thank you for your patience. I'll try! Thanks again!

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

No branches or pull requests

3 participants