Как настроить предварительно обученную модель с нашим собственным набором данных? - PullRequest
0 голосов
/ 28 июня 2019

Я сейчас работаю над Object detection. Я использую Amazon Workspace для обучения моей модели. Я использую модель для detect cars and bikes. Я использовал предварительно обученную модель (то есть Faster-RCNN-Inception-V2 модель) и настроил ее с моим собственным набором данных для 2 меток, а именно автомобиля и велосипеда. Мне потребовалось 5 часов, чтобы закончить обучение. Теперь я хочу изменить мою модель еще на 2 метки (сохраняя старые), а именно на автобус и авто. Но я не хочу проводить обучение с нуля, так как моя модель уже обучена для автомобилей и мотоциклов. Так есть ли способ, которым я могу тренировать свою модель только с набором данных автобуса и авто, и после обучения она обнаружит все 4 objects(car, bike, bus, and auto)?

1 Ответ

0 голосов
/ 29 июня 2019

Загрузить сохраненную модель с предварительно подготовленными весами. Удалите последний плотный слой и слой 2 класса softmax и добавьте новый плотный слой с 4 классами softmax, поскольку ваши функции низкого уровня уже обучены. Теперь обучите эту модель измененным данным.

...