Как понять, является ли тренировка yolov4 переобученной?

Я обучаюсь обнаруживать людей с помощью YOLOv4 в пользовательском наборе данных. Я использовал эту команду для обучения набора данных:

!./darknet detector train data/obj.data cfg/custom-yolov4-detector.cfg yolov4.conv.137 -dont_show -map

В конце обучения он дает такую ​​диаграмму:

введите описание изображения здесь

Проверка дает точность не более 97%. Но когда я наблюдаю за тестовыми данными, это дает точность примерно 80% при видеозаписи. Это переоснащение? Как я могу решить эту проблему? Я думаю, что точность в графике должна расти.


object-detection yolo darknet yolov4 overfitting-underfitting
person John Amell    schedule 12.04.2021    source источник

Ответы (1)


arrow_upward
0
arrow_downward

Это не переоснащение, и неудивительно, что точность на тестовом наборе ниже, чем на проверочном. Модель училась на наборе поездов, который обычно ближе к набору проверки, а также иногда модель настраивается при использовании набора проверки, поэтому ожидается, что модель будет работать лучше на наборе проверки (верно для каждой модели машинного обучения). ).

Переобучение происходит при обучении, если точность набора поездов продолжает расти, в то время как точность проверки уменьшается между эпохами. (переоснащение относится к комплекту поездов).

person Dr.Haimovitz    schedule 12.04.2021
comment
На графике красная линия обозначает точность проверки. В некоторых пунктах он уменьшается. Это нормально? - person John Amell; 12.04.2021
comment
Да, модель учится обобщать любые данные (набор тестов), поэтому это нормально, если для некоторых пакетов ввода она будет работать хуже, чтобы улучшить результаты для теста. - person Dr.Haimovitz; 12.04.2021
comment
Кстати, это не график точности, а другая функция потерь модели. - person Dr.Haimovitz; 12.04.2021
comment
Это тренировочная таблица. Я знаю, что показывает карту и потерю. В источниках, которые я читал, было написано, что валидация mAP должна увеличиваться, а функция потерь - уменьшаться. В составленном мною поезде есть взлеты и падения. - person John Amell; 12.04.2021
comment
Выглядит нормально - person Dr.Haimovitz; 12.04.2021