Могу ли я продолжить тренировку с финального веса с большим количеством тренировочных и тестовых изображений?

Question

Я тренировал свое собственное обнаружение объектов с помощью darknet yolov3, пока средняя потеря не снизилась до 0,06, но теперь я хочу тренировать его с более обучающими и тестовыми изображениями (возможно )также удалив некоторые файлы изображений). Могу ли я выполнить эти шаги и продолжить обучение с окончательным файлом .weights или мне следует начать его с самого начала?

Answer 1

Да, вы можете использовать текущую обученную модель (файл .weights) в качестве предварительно обученной модели для новой тренировки. Например, если вы используете репозиторий AlexeyAB , вы можете обучить свою модель командой, подобной этой:

darknet.exe detector train data/obj.data yolo-obj.cfg darknet53.conv.74

, где darknet53.conv.74 - предварительно обученная модель.

В новой тренировке вы можете добавлять или удалять изображения. Однако основные конфигурации должны быть правильными (например, количество классов и т. Д.).

В соответствии со страницей, которую я упомянул:

в исходном хранилище исходное хранилище файл весов сохраняется только один раз каждые 10 000 итераций