search

WongKinYiu / yolov9

Implementation of paper - YOLOv9: Learning What You Want to Learn Using Programmable Gradient Information
GNU General Public License v3.0
8.79k stars 1.36k forks source link

How to update class names after model has already been trained? #542

Open johanpramudito opened 1 month ago

johanpramudito commented 1 month ago

Is there a way to update the class names for a model after it has been trained?

Let say i train the model, but there are class with name "dgo", the right name is "dog". How to update the class names in YOLOv9?

fukingbus commented 1 month ago

the model treats your dataset as numeric values. no matter what you named inside your class.txt, your bounding box data should begin with numeric class id.

for short, just change dgo to dog inside class.txt and you are good to go 😀