Когда я использую yolov3 для обучения своего набора данных и на изображении более 60 объектов, нужно ли мне менять yolov3.cfg или другие параметры?

Question

Когда я использую yolov3 для обучения моего набора данных, и на изображении более 60 объектов, и эти объекты очень плотные, конечная потеря поезда составляет 16,4, мне нужно изменить yolov3.cfg или другие параметры? Или Что я могу сделать, чтобы уменьшить потерю поезда?

Answer 1

Из вашего вопроса я понимаю, что в ваших изображениях много мелких предметов. Вы можете сделать несколько вещей, чтобы улучшить свои потери при обучении:

• установить флаг random=1 в вашем .cfg-file - это повысит точность, тренируя Йоло для разных разрешений.
• Увеличьте разрешение сети в вашем .cfg-file (height=608, width=608 или любом значении, кратном 32) - это повысит точность.

Также я бы предложил вам иметь несколько тестовых изображений и проверить ваши mAP во время обучения, если точность обнаружения модели увеличивается с уменьшением потерь.