Задать вопрос

Как повысить точность модели YOLO?

Занимаюсь обучением модели по поиску дефектов посадки на одежде на YOLO. Разметку выполняю в Roboflow, всего изображений в датасете изображений 958 (ссылка на проект ), использовал различные методы аугментации изображений, гиперпараметры запуска обучения - model=yolo11m.pt epochs=150 imgsz=960 batch=16 plots=True patience=30 cos_lr=True. Изначально обучал на yolov11s, результаты были не утешительные, поэтому перешёл на yolov11m в надежде на улучшение, но presicion, как и recall остаются на достаточно низком уровне.
Как можно повысить точность модели? Что я делаю не так при обучении?

График обучения:
results.png
  • Вопрос задан
  • 1647 просмотров
Подписаться 4 Простой Комментировать
Помогут разобраться в теме Все курсы
  • Нетология
    Data Scientist: расширенный курс
    13 месяцев
    Далее
  • Яндекс Практикум
    Инженер машинного обучения
    4 месяца
    Далее
  • Skillfactory
    Профессия Data Scientist
    24 месяца
    Далее
Пригласить эксперта
Ответы на вопрос 1
petka_vaska
@petka_vaska
Grow projects
Для мелких и размытых дефектов посадки получаются низкие precision/recall скорее всего не из-за маленькой модели, а из-за шумной разметки, слишком агрессивных аугментаций, несоответствия головы сети масштабу объектов (нужен P2 для мелочи), неверных гиперпараметров, скудных негативных примеров без дефектов (подчеркнуть нужное).
Для начала обрежьте шум. Удалите классы, которые путаете даже вы. Лучше 1–2 устойчивых класса дефект/нет дефекта, чем 5 похожих.
Ответ написан
Комментировать
Ваш ответ на вопрос

Войдите, чтобы написать ответ

Похожие вопросы