Objekterkennung mit Tensorflow

Vor einiger Zeit habe ich eine Objekterkennung AI trainiert

Inhaltsverzeichnis

An einem kalten Wintertag im Juli … das ist in Australien … hatte ich plötzlich den dringenden Bedarf, ein AI-Modell zur Erkennung von unverschlossenen Stahlbetonbewehrungsstäben zu trainieren…

Also habe ich bei lokalen Bunnings einige neue glänzende Reo-Stäbe wie diese gekauft

concrete reinforcement bars

und Sicherheitskappen für Stahlbetonbewehrungsstäbe wie diese

concrete reinforcement bar caps

Mit den Kappen begonnen

Ich habe hundert Fotos unterschiedlicher Kompositionen mit diesen gelben Reo-Kappen gemacht, etikettiert in LabelImg, genommen einige SSD-Modelle, dann ein EfficientNet und habe etwas Programmierung in Python durchgeführt – und das Modell lernte schnell, was ich von ihr wollte.

reo bar cap - lh

reo bar cap - rh

Anfangs erkannte das Modell fast alle gelben zylindrischen Objekte als Reo-Kappen

coffee cap

Und die Umgebungsboxen könnten präziser sein…

Dann kamen die Stäbe

Ich habe mehrere andere nützliche Werkzeuge aus der Garage genommen und hundert weitere Fotos wie diese gemacht, und das Modell wurde besser…

garage tools object detection

Definitiv besser

garage tools object detection

ai object detection training in the gerage

Mobile App

Schließlich trainierte ich ein kleines SSD-AI-Modell, das auf einem Mobiltelefon ohne große Latenz laufen kann, erstellte eine einfache Android-App und so sah sie aus:

mobile phone ssd ai

mobile phone object detection

mobile tensorflow

Richtig.