Как повторно тренировать нестандартные веса йоло?

Я выполнил пользовательское обнаружение в YOLOv3 более трех классов, но обнаружение не было точным, поэтому я хочу переобучить свои пользовательские веса YOLO с большим количеством изображений, но когда я запускаю его с новыми изображениями, он сразу же завершается, что я сделал неправильный?

Вот как я это тренирую

 !./darknet detector train data/obj.data cfg/yolov3_custom.cfg yolov3_custom_last.weights

Содержание obj.data:

classes = 3
train = data/train.txt
valid = data/test.txt
names = data/obj.names
backup = /mydrive/yolov3/backup/

Содержание yolov3_custom.cfg:

# Training
batch=64
subdivisions=16
width=416
height=416
channels=3
momentum=0.9
decay=0.0005
angle=0
saturation = 1.5
exposure = 1.5
hue=.1
learning_rate=0.001
burn_in=1000
max_batches = 6000
policy=steps
steps=4800,5400
scales=.1,.1

person s683    schedule 01.07.2020    source источник
comment
Требуется дополнительная информация. Каково содержимое файлов obj.data и yolov3_custom.cfg?   -  person AbdelAziz AbdelLatef    schedule 02.07.2020
comment
@AbdelAzizAbdelLatef привет, проверьте правку. Благодарю.   -  person s683    schedule 02.07.2020
comment
Какая у вас операционная система?   -  person AbdelAziz AbdelLatef    schedule 03.07.2020
comment
1. Я думаю, что должен быть какой-то журнал ошибок, если его останавливает какая-то проблема. Можете ли вы подтвердить, закончилось ли обучение с журналом ошибок или нет? 2. Какое значение max_batches вы установили для 1-й тренировки и 2-й тренировки? 3. Используете ли вы вес последней итерации 1-й тренировки для 2-й тренировки?   -  person Hamid Shatu    schedule 03.07.2020
comment
@HamidShatu 1. ошибки нет, 2. max_batches = 6000 для 1-го и 2-го, 3. да, я использую последние веса, которые достигают максимальных пакетов, поэтому он остановлен? я должен изменить max_batches, чтобы он был удвоен по сравнению с первой тренировкой? спасибо за ответ   -  person s683    schedule 03.07.2020
comment
@ s683 Я предполагаю, как вы догадались, поскольку max_batches для обеих тренировок одинаковы, и вы используете последний вес из 1-й тренировки, поэтому тренировка заканчивается, как только вы ее начинаете. Да, пожалуйста, удвойте значение max_batches и снова запустите поезд и посмотрите, что произойдет.   -  person Hamid Shatu    schedule 03.07.2020


Ответы (2)


Просто измените путь к файлу весов в команде для обучения модели и запустите ее снова. Нет необходимости обновлять параметр max_batches или каким-либо образом изменять файл конфигурации. Фреймворк знает, для скольких итераций был обучен данный набор весов при повторном обучении модели. Например, если вы должны остановиться после 2000 итераций в начальном процессе обучения, при повторном обучении с использованием набора весов, который был создан в результате, он начнется на 2001-й итерации.

person Minura Punchihewa    schedule 25.12.2020

@ Хамид Шату. Ты прав. Вам нужно обновить max_batches =, а также проверить имя файла веса.

person user14132572    schedule 19.08.2020
comment
опубликуйте это как комментарий вместо ответа! - person Shobhit; 19.08.2020