Обнаружение объектов с использованием TensorFlow

Несколько времени назад я обучил объектный детектор ИИ

Содержимое страницы

На одном холодном зимнем дне в июле… то есть в Австралии… у меня возникла срочная необходимость обучить модель ИИ для обнаружения неснабжённых арматурных стержней…

Итак, я купил в местном Bunnings некоторые новые блестящие арматурные стержни, такие как эти

concrete reinforcement bars

и защитные колпачки для арматуры, такие как эти

concrete reinforcement bar caps

Начало с колпачков

Сфотографировал сотню изображений с различными композициями этих жёлтых колпачков для арматуры, пометил их в LabelImg, взял SSD модель, затем ещё одну EfficientNet и немного написал код на Python- и модель быстро освоила то, что я от неё хотел.

reo bar cap - lh

reo bar cap - rh

изначально модель принимала почти все жёлтые цилиндрические объекты за заглушки для арматуры

coffee cap

И контурные прямоугольники могли быть точнее…

Потом пришли стержни

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

garage tools object detection

Определённо лучше

garage tools object detection

ai object detection training in the gerage

Мобильное приложение

Наконец обучил небольшую SSD модель ИИ, чтобы она работала на мобильном телефоне без большой задержки, создал простое Android приложение и вот как оно выглядело:

mobile phone ssd ai

mobile phone object detection

mobile tensorflow

Правильно.

Полезные ссылки