Как обучить модель обнаружения объекта новому классу с пользовательскими данными?
Я хочу обучить модель yolov3 двум классам: Person и dustbin.Так как есть веса, которые уже тренированы в 80 различных классах, включая людей, я хочу использовать эту предварительно обученную модель, чтобы тренироваться на моем мусорном ящике нового класса.Я удалил ~ 1000 изображений мусорных баков и пометил их.У меня путаница.
- Поскольку модель уже обучена для человека, нужно ли предоставлять ей помеченные данные категории человека?Или я могу просто использовать данные, помеченные
Dustbin, и они (надеюсь) начнут обнаруживать мусорные баки и людей (например, базовую модель)
Я уже внес изменения в файлы coco.names, которые выглядяткак
person
dustbin
, а также внес необходимые изменения в файл cfg, например, фильтр на 21.
Извините за наивные вопросы, я только начал изучать обнаружение объектов и yolo и старался изо всех силотформатировать вопрос в соответствии с требованиями stackoverflow.Спасибо