Detecção de Objetos com TensorFlow

Há bastante tempo, treinei um sistema de IA de detecção de objetos.

Conteúdo da página

Num dia frio de julho … que é na Austrália … Senti uma necessidade urgente de treinar um modelo de IA para detectar barras de aço de reforço de concreto sem as tampas de proteção…

Então, comprei na loja local Bunnings Warehouse algumas barras de aço novas e brilhantes, como estas

barras de reforço de concreto

e tampas de segurança para as barras, como estas

tampas de barras de reforço de concreto

Começando com as tampas

Tirei cem fotos de diferentes composições com essas tampas amarelas de barras de aço, etiquetadas no LabelImg, peguei um modelo SSD, depois outro EfficientNet, fiz um pouco de programação em [Python](https://www.glukhov.org/pt/tags/python/ “artigos sobre Python)- e o modelo rapidamente entendeu o que eu queria dela.

tampa de barra de aço - mão esquerda

tampa de barra de aço - mão direita

Inicialmente, o modelo considerava quase todos os objetos cilíndricos amarelos como tampas de barras de aço

tampa de café

E as caixas delimitadoras poderiam ser mais precisas…

Depois vieram as barras

Peguei várias outras ferramentas úteis da garagem e tirei outras cem fotos como esta, e o modelo começou a ver melhor…

detecção de objetos de ferramentas de garagem

Sim, definitivamente melhor.

detecção de objetos de ferramentas de garagem

treinamento de detecção de objetos de IA na garagem

Aplicativo Mobile

Finalmente, treinei um pequeno modelo de IA SSD para rodar no meu celular sem latência enorme, criei um aplicativo Android simples e é assim que ele ficou no Samsung S8:

celular SSD IA

detecção de objetos em celular

tensorflow mobile

Foi um experimento muito bom.