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

Unable to build engine file for Yolov7 Custom model #1528

Open
Muthukumar4796 opened this issue May 20, 2024 · 1 comment
Open

Unable to build engine file for Yolov7 Custom model #1528

Muthukumar4796 opened this issue May 20, 2024 · 1 comment

Comments

@Muthukumar4796
Copy link

Muthukumar4796 commented May 20, 2024

Env

  • GPU RTX3050
  • OS, Ubuntu 22.04
  • Cuda & TRT version I have uploaded image
  • Yolov7 w6 custom model
  • image size H:768 W:1280 and Num of classes : 6

I'm having trouble building the engine file for my custom YOLOv7 model. I've made changes to the config.h file for classes and image size, but I still can't build the engine file. However, when I tried using the YOLOv7 w6 pretrained model, the engine file was created successfully.

Screenshot from 2024-05-20 12-16-34

Screenshot from 2024-05-20 15-09-58

Screenshot from 2024-05-20 12-26-17

@wang-xinyu
Copy link
Owner

TensorRT 10.x is not supported yet.

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

2 participants