Objekterkennung mit TensorFlow

Vor einiger Zeit habe ich eine Objekterkennungs-KI trainiert

Inhaltsverzeichnis

An einem kalten Wintertag im Juli… das ist in Australien… verspürte ich ein dringendes Bedürfnis, ein KI-Modell zum Erkennen von unverkapselten Stahlbewehrungsstäben zu trainieren…

Also kaufte ich in einem lokalen Bunnings Warehouse einige neue glänzende Bewehrungsstäbe wie diese

concrete reinforcement bars

und Sicherheitskappen für die Stäbe wie diese

concrete reinforcement bar caps

Begann mit den Kappen

Machte hundert Fotos mit diesen gelben Bewehrungskappen in verschiedenen Kompositionen, beschriftet in LabelImg, nahm ein SSD-Modell, dann ein anderes EfficientNet, machte etwas Programmierung in Python - und das Modell erfasste schnell, was ich von ihr wollte.

reo bar cap - lh

reo bar cap - rh

Anfangs erkannte das Modell fast alle gelben zylindrischen Objekte als Bewehrungskappen

coffee cap

Und die Begrenzungsrahmen könnten präziser sein…

Dann kamen die Stäbe

Nahm einige andere nützliche Werkzeuge aus der Garage und machte noch einmal hundert Fotos wie diese, und das Modell wurde besser…

garage tools object detection

Ja, definitiv besser

garage tools object detection

ai object detection training in the gerage

Mobile App

Schließlich trainierte ich ein kleines SSD-KI-Modell, das auf einem Mobiltelefon ohne große Latenz läuft, erstellte eine einfache Android-App und hier ist, wie es auf einem Samsung S8 aussah:

mobile phone ssd ai

mobile phone object detection

mobile tensorflow

Richtig.