How to modify names of a YOLOv5 trained

[GiorgioSgl]

❔Question

I run the command
python3 detect.py --source image.jpg --weights runs/train/exp5/last.pt

The problem is that I mismatch some name of the classes and I have to change it, but seems impossible.

How can I change it??
I need it asap, that I have to submit my thesis :)

Additional context

I try already changing the array in the .yaml file but no success :(

[GiorgioSgl]

Just solve in a hard way, I don't know if there are smoother solutions

Actually here

yolov5/detect.py

Line 59 in 095197b

names = model.module.names if hasattr(model, 'module') else model.names # get class names

You have to modify in the following way

names = ["myfirstclass","my second class",..,"last class"]

Hope it help and if there are better solution, just let me know

[bkbilly]

I had the same question, but I solved in a similar way like this:
After loading the model:

yolov5/detect.py

Line 56 in 095197b

model = attempt_load(weights, map_location=device) # load FP32 model

I replaced the dictionary like so:

model.names[0] = 'first_class'
model.names[1] = 'second_class'
model.names[2] = 'third_class'