Deteksi Objek Dengan Tensorflow

Beberapa waktu yang lalu saya melatih AI detektor objek

Konten Halaman

Pada satu hari musim dingin di bulan Juli… yang berada di Australia… saya merasa ada kebutuhan mendesak untuk melatih model AI dalam mendeteksi batang reinforcement beton yang tidak terkait…

Jadi saya membeli dari toko lokal Bunnings Warehouse beberapa batang reo baru yang berkilau seperti ini

concrete reinforcement bars

dan penutup keamanan batang seperti ini

concrete reinforcement bar caps

Mulai dengan penutup

Saya mengambil seratus foto dengan komposisi berbeda menggunakan penutup reo berwarna kuning ini, dilabelkan dengan LabelImg, menggunakan model SSD, lalu EfficientNet dan melakukan sedikit coding dalam Python- dan model dengan cepat mempelajari apa yang saya inginkan darinya.

reo bar cap - lh

reo bar cap - rh

Awalnya model menganggap hampir semua objek silinder berwarna kuning sebagai penutup batang reo

coffee cap

Dan batas kotak bisa lebih presisi…

Lalu batang datang

Saya mengambil beberapa alat berguna lainnya dari garasi dan mengambil seratus foto seperti ini, dan model mulai melihat lebih baik…

garage tools object detection

Ya, jelas lebih baik.

garage tools object detection

ai object detection training in the gerage

Aplikasi Mobile

Akhirnya, saya melatih model SSD AI kecil untuk menjalankannya di ponsel saya tanpa latensi besar, membuat aplikasi Android sederhana dan berikut tampilannya di Samsung S8:

mobile phone ssd ai

mobile phone object detection

mobile tensorflow

Ini adalah eksperimen yang sangat baik.

Tautan yang Berguna