tensorrtx
tensorrtx copied to clipboard
Unable to build engine file for Yolov7 Custom model
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.
TensorRT 10.x is not supported yet.
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.