Как использовать класс «ignore» с API обнаружения объектов tenorflow?

Question

Я обучил модель обнаружения объекта тензорного потока (для num_steps:50000), используя SSD (mobilenet-v1) в пользовательском наборе данных. Я получил mAP@.50IOU ~ 0,98 и loss ~ 1,17. Набор данных состоит из нескольких изображений игральных карт (пропустить, повернуть вспять и нарисовать четыре). На всех этих картах модель работает довольно хорошо, так как я обучил модель только на этих 3 картах (около 278 изображений с 829 ограничивающими рамками (25% ограничивающих рамок, используемых для тестирования, т.е. проверки), собранных с помощью мобильного телефона).

Тем не менее, я не тренировал модель на других картах, но все же он обнаруживает другие карты (вывод с помощью веб-камеры).

Как я могу это исправить? Должен ли я также собирать изображения других классов (что-либо кроме пропуска, реверсирования и вытягивания четырех карт) и игнорировать этот класс в действии? Таким образом, эта модель видит этот класс, т. Е. Метка: другие изображения во время обучения и не ставит никаких меток во время вывода.

Я не уверен, как сообщить API обнаружения объекта tenorflow, что он должен игнорировать изображения из Другие класс.

Кто-нибудь может предоставить указатель?

Пожалуйста, поделитесь своими взглядами!

Answer 1

Да , вам нужен другой класс, который является объектом , который вы не хотите обнаруживать.

Если у вас нет этого Другой класс , который включает в себя все, что не должно быть обнаружено. Модель сравнит его с существующим классом, который практически идентичен интересующим картам.

Некоторые факторы:

1. Сходство Форма
2. Сходство Цвет
3. Сходство Символы

Вот почему, хотя это не карта интереса (Skip, Reverse и Draw 4), он каким-то образом будет иметь высокую «принадлежность» к этим трем классам.

Наличие другого класса для сброса всех этих элементов может значительно уменьшить «принадлежность» к трем интересующим классам. и как можно больше предоставьте ЛОТ Данных во время обучения.

Если вы не хотите иметь другой класс.

Вы можете переодеться Пропустите, поменяйте местами и возьмите 4 карты (около 100%), затем увеличьте пороговое значение обнаружения до (70-90%).

Надеюсь, это поможет вам.