Как использовать класс из MS-COCO и добавить 2 новых класса с помощью маски RCNN? - PullRequest
0 голосов
/ 08 марта 2020

Я хочу, чтобы моя модель распознавала только 3 класса, т.е. человека, шины и дороги. Я использую предварительно подготовленные гири MS COCO, которые уже прошли обучение на уроке «Персона».

Как использовать этот уже существующий класс «Персона» и дополнительно обучить мою модель 2 новым классам, т. Е. «Шина» и «Дорога»?

Я установил NUM_CLASSES = 1 + 3 (фон + objects).

Однако в моем наборе данных есть только аннотации для 'Tire' и 'Road', и, следовательно, это единственные классы, которые я добавляю, используя self.add_class(source_name, class_id, class_name).

...