Как добавить класс в существующую модель? - PullRequest
2 голосов
/ 14 июня 2019

Я обучил модель, использующую обнаружение объектов tenorflow / SSD mobilenet. Отлично работает!

Я бы хотел добавить к нему класс - просто чтобы обнаружить ручки или что-то в этом роде.

Как я могу это сделать?

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

Спасибо!

Ответы [ 2 ]

2 голосов
/ 14 июня 2019

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

Есть несколько хороших руководств по созданию пользовательского детектора объектов с использованием API обнаружения объектов tenorflow.

Например, sentdex опубликовал очень хорошее пошаговое руководство здесь . Также на официальной странице репозитория github есть несколько хороших учебных пособий, таких как этот: добавление вашего собственного набора данных , на самом деле это то же самое, что добавить или удалить классы из предварительно обученной модели в некотором смысле.

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

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

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

...