Rilevamento oggetti con Tensorflow

Un po' di tempo fa ho addestrato un riconoscitore oggetti AI

Indice

Un freddo giorno d’inverno a luglio… che è in Australia… ho avuto un forte bisogno di addestrare un modello AI per rilevare barre di armatura in calcestruzzo non chiuse…

Così ho comprato da un negozio locale Bunnings Warehouse alcune nuove barre di armatura lucide come queste

barre di armatura in calcestruzzo

e coperture per barre di armatura come queste

coperture per barre di armatura in calcestruzzo

Iniziato con le coperture

Ho scattato centinaia di foto di diverse composizioni con queste coperture gialle per barre di armatura, etichettate in LabelImg, ho utilizzato un modello SSD, poi un EfficientNet e ho fatto un po’ di coding in Python- e il modello ha rapidamente imparato ciò che volevo da lei.

copertura per barra di armatura - lh

copertura per barra di armatura - rh

inizialmente il modello ha riconosciuto quasi tutti gli oggetti cilindrici gialli come coperture per barre di armatura

copertura per caffè

E i bounding box potrebbero essere più precisi…

Poi sono arrivare le barre

Ho preso diversi altri strumenti utili dal garage e ho scattato altre centinaia di foto come questa, e il modello ha iniziato a vedere meglio…

rilevamento oggetti in garage

Sì, decisamente meglio.

rilevamento oggetti in garage

addestramento AI per rilevamento oggetti nel garage

App Mobile

Infine, ho addestrato un piccolo modello SSD AI per eseguirlo sul mio telefono mobile senza grandi ritardi, ho creato un’app Android semplice e ecco come appariva sul Samsung S8:

telefono mobile SSD AI

rilevamento oggetti su telefono mobile

TensorFlow su telefono mobile

È stato un esperimento molto utile.