search

cyrusbehr / YOLOv8-TensorRT-CPP

YOLOv8 TensorRT C++ Implementation
MIT License
568 stars 70 forks source link

How to export custom trained engine file #45

Closed IamShubhamGupto closed 9 months ago

IamShubhamGupto commented 9 months ago

Hello

I have a custom trained pt file of yolov8 and when I create an engine file using your codebase, it defaults to the coco dataset classes. How do I export and run an engine file which is trained on my data?

Thank you for your help

suryan-s commented 9 months ago

Hello

I have a custom trained pt file of yolov8 and when I create an engine file using your codebase, it defaults to the coco dataset classes. How do I export and run an engine file which is trained on my data?

Thank you for your help

Same query here!!

cyrusbehr commented 9 months ago

You have two options for specifying your own classes.

1) Change the class names in the header file here

2) Use the --class-names command line argument.