tensorrtx icon indicating copy to clipboard operation
tensorrtx copied to clipboard

Published 20 hours ago • verified publisher icon wang-xinyu

Unable to build engine file for Yolov7 Custom model

Open Muthukumar4796 opened this issue 9 months 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