Как выполнить заданное c распознавание объекта на изображении?

Question

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

Сначала я попытался выполнить распознавание объектов, используя нейронную сеть, такую ​​как маска R-CNN с python и тензор потока. Но я понял, что у меня нет огромного тренировочного набора, только мои 3 изображения. Алгоритмы нейронной сети, похоже, адаптированы для распознавания таких понятий, как собака, смартфон, пейзаж, но не конкретная собака, конкретный смартфон c или конкретный пейзаж c.

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

Какие алгоритмы лучше всего подходят для такого распознавания?

Answer 1

Попробуйте использовать набор данных COCO . Поскольку весовые коэффициенты COCO уже были обучены на тысячах предметов и изображений, вы просто сможете запустить функцию splash, чтобы помочь обнаружению с помощью маски RCNN.

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