Detección de objetos con Tensorflow

Un frío día de invierno en julio… que es en Australia… sentí la urgente necesidad de entrenar un modelo de IA para detectar barras de refuerzo de concreto sin tapones…

Así que compré en la tienda local de Bunnings Warehouse algunas nuevas barras de refuerzo brillantes como estas

y tapones de seguridad para barras de refuerzo como estos

Comencé con los tapones

Tomé cien fotos de diferentes composiciones con estos tapones amarillos de barras de refuerzo, etiquetadas en LabelImg, tomé un modelo SSD, luego otro EfficientNet y realicé un poco de programación en Python- y el modelo rápidamente aprendió lo que quería de ella.

inicialmente el modelo tomó casi todos los objetos cilíndricos amarillos como tapones de barras de refuerzo

Y los cuadros delimitadores podrían ser más precisos…

Luego llegaron las barras

Tomé varios otros útiles de herramientas del garaje y tomé otras cien fotos como esta, y el modelo comenzó a ver mejor…

Sí, definitivamente mejor.

Aplicación móvil

Finalmente, entrené un pequeño modelo SSD de IA para ejecutarlo en mi teléfono móvil sin una gran latencia, creé una aplicación Android simple y aquí es cómo se veía en un Samsung S8:

Fue un experimento muy bueno.

Enlaces útiles

• MMdetection ya no está soportado
• Flujo de trabajo para entrenar un detector de objetos con Label Studio & MMDetection
• LabelImg: https://github.com/HumanSignal/labelImg
• TensorFlow: https://www.tensorflow.org/
• Hoja de trucos de Python
• Hoja de trucos de Conda
• Hoja de trucos de venv
• Hoja de trucos de Bash
• Hoja de trucos de Ollama
• Generando PDF en Python - Bibliotecas y ejemplos"