Я только что закончил обучение YOLOv3 в даркнете на моем собственном наборе данных, в котором было всего 100 изображений. Теперь я хочу обучить его работе с большим набором данных (500 изображений). Мне было интересно, могу ли я использовать эту предварительно обученную модель для обучения на моем новом наборе данных, не начиная с самого начала. Кроме того, будет ли он тренироваться быстрее, чем раньше?
Снова обучите даркнет на большем наборе данных
Ответы (2)
Я думаю, вы можете использовать предварительно натренированные веса, чтобы продолжить тренировку с новым набором изображений. Проверьте это Как тренироваться крошечный-йоло.
person
AbdelAziz AbdelLatef
schedule
22.09.2019
Вы можете сделать что-то подобное, просто продолжите обучение с последней сохраненной моделью. Что-то вроде
./darknet detector train data/obj.data yolo-obj.cfg yolov4.conv.137
Замените yolov4.conv.137 вашей последней сохраненной моделью.
./darknet detector train data/obj.data yolo-obj.cfg backup/yolov3_last.weights
Вы должны изменить свои max_batches в вашем cfg, потому что ваше обучение продолжится с того места, где оно закончилось.
И, возможно, ваша модель превратится в лучшую модель быстрее, чем раньше, но поскольку вы загружаете в 5 раз больше изображений, это может быть медленнее.
person
Rafael Junio Xavier
schedule
11.09.2020