Rilevamento oggetti con Tensorflow

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 Bunnings qualche nuova barra di armatura lucida come queste

e coperture di sicurezza per le barre come queste

Iniziato con le coperture

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

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

E i bounding box potrebbero essere più precisi…

Poi sono arrivate le barre

Ho preso diversi altri strumenti utili dal garage e ho scattato altre centinaia di foto come questa, e il modello è migliorato…

Definitivamente migliore

App Mobile

Infine ho addestrato un piccolo modello SSD AI per funzionare su un telefono mobile senza grandi ritardi, ho creato un’app Android semplice e ecco come appariva:

Giusto.

Link utili

• MMdetection non è più supportato
• Flusso di lavoro per addestrare un rilevatore oggetti con Label Studio & MMDetection
• LabelImg: https://github.com/HumanSignal/labelImg
• TensorFlow: https://www.tensorflow.org/
• Python Cheatsheet
• Conda Cheatsheet
• venv Cheatsheet
• Bash Cheat Sheet
• Ollama cheatsheet
• Docker Cheatsheet
• Generazione PDF in Python - Librerie ed esempi"