Дообучение YOLO V8
Можно ли дообучить уже обученную модель YOLO V8?
Обучал на пользовательских данных пару дней. Хочется добавить в уже обученную модель обнаружение других объектов, и не очень хочется тратить лишнее время на обучение модели тому, что она и так знает.
Подскажите, как мне это сделать?
Ответы (2 шт):
Данный вопрос гуглится по первой ссылке в интернете, вот к примеру отличная статья с хабра: https://habr.com/ru/articles/714232/
Официальная документация: https://docs.ultralytics.com/ru/modes/train/
Почитай тут: https://docs.ultralytics.com/modes/train/#how-do-i-resume-training-from-an-interrupted-session-in-ultralytics-yolo11, раздел "How do I resume training from an interrupted session in Ultralytics YOLO11?"
Пример кода на Python:
from ultralytics import YOLO
# Load the partially trained model
model = YOLO("path/to/last.pt")
# Resume training
results = model.train(resume=True)
И тот же код на CLI:
yolo train resume model=path/to/last.pt
Такой же метод дообучения применим и к другим моделям YOLO)
Так же можете попробовать такой метод:
if __name__ == "__main__":
from ultralytics import YOLO
model = YOLO("path/to/last.pt")
model.train(data="data.yaml", epochs=125)
metrics = model.val()