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 loja local Bunnings algumas novas barras de reo brilhantes como estas

concrete reinforcement bars

e tampas de segurança para barras de reforço como estas

concrete reinforcement bar caps

Comecei com as tampas

Fiz cem fotos de diferentes composições com estas tampas amarelas de reo bar, rotuladas em 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 depósito e fiz outras cem fotos como esta, e o modelo ficou melhor…

garage tools object detection

Definitivamente melhor

garage tools object detection

ai object detection training in the gerage

Aplicativo Mobile

Finalmente treinei um pequeno modelo SSD de IA para rodar em um telefone móvel sem grande latência, criei um aplicativo Android simples e aqui está como ele ficou:

mobile phone ssd ai

mobile phone object detection

mobile tensorflow

Certo.