Objectdetectie 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 betonversterkbalken…

Dus ik koopde bij de lokale Bunnings Warehouse winkel enkele nieuwe glanzende reo-balken zoals deze

concrete reinforcement bars

en balkveiligheidskappen 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, en een EfficientNet en deed wat code 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 grensboxen konden preciezer zijn…

Dan kwamen de balken

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

garage tools object detection

Ja, zeker beter.

garage tools object detection

ai object detection training in the gerage

Mobiele app

Uiteindelijk trainde ik een klein SSD AI-model om het op mijn mobiele telefoon te laten draaien zonder grote latentie, maakte een eenvoudige Android-app en hier is hoe het eruitzag op een Samsung S8:

mobile phone ssd ai

mobile phone object detection

mobile tensorflow

Het was een zeer goede experiment.