Object Detectie met Tensorflow

Een tijd geleden heb ik een objectdetectie AI getraind

Inhoud

Op een koude winterdag in juli … dat is in Australië … had ik plotseling het gevoel om een AI-model te trainen voor het detecteren van ongedekte betonversterkingsstaven…

Dus ik koop bij de lokale Bunnings enkele nieuwe glanzende reo-staven zoals deze

concrete reinforcement bars

en veiligheidskappen voor staven zoals deze

concrete reinforcement bar caps

Begonnen met kappen

Ik nam honderd foto’s van verschillende samenstellingen met deze gele reo-kappen, gemarkeerd in LabelImg, nam een SSD-model, dan een EfficientNet en deed wat coding in Python- en het model leerde snel wat ik van haar wilde.

reo bar cap - lh

reo bar cap - rh

aanvankelijk nam het model vrijwel alle gele cilindrische objecten als reo-kappen

coffee cap

En de randen van de bounding boxes konden preciezer zijn…

Dan kwamen de staven

Ik nam verschillende andere nuttige gereedschappen uit de garage en nam nog honderd foto’s zoals deze, en het model werd beter…

garage tools object detection

Definitief beter

garage tools object detection

ai object detection training in the gerage

Mobiele app

Uiteindelijk getraind een klein SSD AI-model om op een mobiele telefoon te draaien zonder grote vertraging, gemaakt van een eenvoudige Android-app en zo zag het eruit:

mobile phone ssd ai

mobile phone object detection

mobile tensorflow

Juist.