Objektdetektion med TensorFlow

På en kall vinterdag i juli… det är i Australien… kände jag ett brådskande behov att träna en AI-modell för att detektera oskyddade betongförstärkande stänger…

Så köpte jag från den lokala Bunnings Warehouse butiken några nya glansiga reo-stänger som dessa

och säkerhetslock för stänger som dessa

Började med locken

Tog hundra bilder av olika kompositioner med dessa gula reo-stängslock, märkte i LabelImg, tog en SSD-modell, sedan en annan EfficientNet gjorde lite kodning i Python - och modellen lärde sig snabbt vad jag ville av henne.

initialt tog modellen nästan alla gula cylindriska föremål som reo-stängslock

Och begränsningsrutorna kunde vara mer precisa…

Sedan kom stängerna

Tog flera andra användbara verktyg från garaget och tog ytterligare hundra bilder som dessa, och modellen började se bättre…

Ja, definitivt bättre.

Mobilapp

Slutligen tränade jag en liten SSD AI-modell för att köra den på min mobiltelefon utan stor latens, skapade en enkel Android-app och här är hur den såg ut på Samsung S8:

Det var ett mycket bra experiment.

Användbara länkar

• MMdetection stöds inte längre
• Arbetsflöde för träning av objektdetektor med 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
• Generering av PDF i Python - Bibliotek och exempel