YoloV5 Индивидуальная переподготовка

Question

Я обучил свой собственный набор данных модели yoloV5s, и мои выводы достигли 80% точности. Теперь мне нужно повысить точность, добавив больше изображений и этикеток.

Мой вопрос в том, что я уже обучил 10000+ лейблов, чтобы достичь 80%, мне потребовалось 7 часов. нужно ли мне включать старые 10000+ данных с моими новыми данными, которых всего 1000, для обучения и повышения моей точности?

Есть ли способ включить новые данные только для переобучения модели, даже если я добавляю новый класс?

Пожалуйста, помогите здесь, чтобы я мог сэкономить свое время и место.

Answer 1

Вопрос, который вы задаете, касается непрерывного изучения темы , которая в настоящее время является активной областью исследований. Поскольку вам нужно добавить больше классов в вашу модель, вам нужно добавить новый класс с предыдущими данными и заново обучить модель с самого начала. Если вы этого не сделаете, т.е. вы тренируетесь только в новом классе, ваша модель полностью забудет о предыдущих данных (изученная функция); это забвение известно как катастрофическое забвение .

Многие люди предлагали различные способы избежать этого Катастрофического забывания; Я лично считаю, что прогрессивная нейронная сеть очень невосприимчива к забыванию. Помимо этого, вы можете найти другие методы.здесь

Как я уже говорил, в настоящее время это очень активная область исследований; Полноценного решения не существует. На данный момент лучший способ - добавить новые данные к предыдущим и переобучить вашу модель.