Пользовательское обнаружение объектов с помощью API Tensorflow Object Detection
Я хочу обнаружить шлем и человека, не носящего шлем.. Я использую модель ssd_mobilenet_v1_coco_11_06_2017 с 1000 изображений шлема и 1000 без шлема.
1) Можно ли использовать модель ssd_mobilenet_v1_coco_11_06_2017 для обучения? Если нет, то какую модель я могу использовать? (потому что у COCO есть только 90 классов и шлем не в этом списке)
Может ли кто-нибудь помочь мне очистить мои сомнения...
1 ответ
Решение
Обратитесь по следующим ссылкам для достижения этой цели
https://pythonprogramming.net/training-custom-objects-tensorflow-object-detection-api-tutorial/
https://medium.com/@dana.yu/training-a-custom-object-detection-model-41093ddc5797