tensorrtx icon indicating copy to clipboard operation
tensorrtx copied to clipboard

Unable to build engine file for Yolov7 Custom model

Open Muthukumar4796 opened this issue 1 year ago • 2 comments

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

Muthukumar4796 avatar May 20 '24 06:05 Muthukumar4796

TensorRT 10.x is not supported yet.

wang-xinyu avatar May 30 '24 03:05 wang-xinyu

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.

stale[bot] avatar Sep 06 '24 02:09 stale[bot]