CHANGING NAME OF PREDICTED LABEL ON PREDICTED IMAGE

[edanurkamaci]

Hello i have an urgent question. I do not have time to train a model again so i wanted to change the name of labels after test result, but i want new labels on predicted result image. How can i do that?

[dh031200]

@edanurkamaci

You can change the name using the code below.
This is the person -> human example.

import torch
from ultralytics import YOLO

model = YOLO('yolov8n.pt')

print('Before: ')
print(model.model.names)
print()


indices = list(model.model.names.values())

# Change name here
## model.model.names[indices.index(BEFORE)] = AFTER
model.model.names[indices.index('person')] = 'human'

print('After: ')
print(model.model.names)


ckpt = {
    'epoch': model.ckpt['epoch'],
    'best_fitness': model.ckpt['best_fitness'],
    'model': model.ckpt['model'],
    'ema': model.ckpt['ema'],
    'updates': model.ckpt['updates'],
    'optimizer': model.ckpt['optimizer'],
    'train_args': model.ckpt['train_args'],
    'date': model.ckpt['date'],
    'version': model.ckpt['version']}

torch.save(ckpt, 'rename.pt')

Result

Before:

After:

[edanurkamaci]

Thanks a lot. It worked

[glenn-jocher]

@edanurkamaci i'm glad to hear that it worked for you! If you have any more questions or need further assistance, feel free to reach out. Happy detecting! 😊🚀