Detecção de Objetos com Tensorflow

Há algum tempo atrás, treinei um detector de objetos AI

Conteúdo da página

Num frio dia de inverno em julho… que é na Austrália… senti a necessidade urgente de treinar um modelo de IA para detectar barras de reforço de concreto não vedadas…

Então comprei em uma loja local de Bunnings Warehouse algumas novas barras de reo brilhantes como estas

concrete reinforcement bars

e tampas de segurança de barra como estas

concrete reinforcement bar caps

Comecei com as tampas

Tirei cem fotos de diferentes composições com estas tampas amarelas de reo bar, rotuladas no LabelImg, usei um modelo SSD, depois outro EfficientNet e fiz um pouco de programação em Python- e o modelo rapidamente aprendeu o que eu queria dela.

reo bar cap - lh

reo bar cap - rh

inicialmente o modelo considerava quase todos os objetos cilíndricos amarelos como tampas de reo bar

coffee cap

E as caixas delimitadoras poderiam ser mais precisas…

Depois vieram as barras

Peguei vários outros úteis ferramentas do garagem e tirei outra centena de fotos como esta, e o modelo começou a enxergar melhor…

garage tools object detection

Sim, definitivamente melhor.

garage tools object detection

ai object detection training in the gerage

Aplicativo Mobile

Finalmente, treinei um pequeno modelo SSD de IA para executá-lo no meu telefone celular sem grande latência, criei um aplicativo Android simples e aqui está como ele parecia no Samsung S8:

mobile phone ssd ai

mobile phone object detection

mobile tensorflow

Foi um excelente experimento.