1 minute read

Week 24 – Balanceo del dataset y entrenamiento con PilotNet/MobileNet

Esta semana se trabajó en el balanceo del dataset de conducción y en el entrenamiento de dos modelos (Trenet/PilotNet y MobileNet) usando únicamente muestras balanceadas.

Avances

  • Se analizaron las distribuciones originales de steer y throttle, detectando una fuerte concentración en conducción recta con acelerador alto.
  • Se aplicó un balanceo del dataset, reduciendo las muestras de conducción recta y manteniendo más ejemplos de giro y frenado reduciendo un dataset de 30000 muestras a 10000.
  • Se entrenaron los modelos Trenet (PilotNet) y MobileNet usando un subconjunto balanceado de 10 000 muestras, con las mascaras del camino.
  • Se realizaron pruebas en el simulador para observar el comportamiento de los modelos entrenados con el nuevo dataset balanceado y sigue el carril en recta correctamente.

Pruebas realizadas

  • Entrenamiento de Trenet/PilotNet con 10 000 muestras balanceadas y evaluación cualitativa en CARLA.
  • Entrenamiento de MobileNet con las mismas 10 000 muestras balanceadas para comparar estabilidad y capacidad de mantener el carril.
  • Ejecución de escenarios de simulación controlados para observar diferencias entre el modelo entrenado con el dataset original y el modelo entrenado con el dataset balanceado.

Próximos pasos

  • Refinar el criterio de balanceo (umbrales de steer, throttle y brake) para mejorar aún más la representación de situaciones críticas.
  • Comparar cuantitativamente Trenet vs MobileNet (métricas de error en validación y tiempo hasta salirse del carril en simulación).

Distribución original de Steer vs Throttle
Distribución original de steer y throttle antes del balanceo.
Distribución balanceada de Steer vs Throttle
Distribución de steer y throttle después del balanceo (10 000 muestras).
Simulación con modelo entrenado en dataset balanceado
Pruebas en el simulador usando modelos entrenados con el nuevo dataset balanceado.
Mascara de recorte
Mascara de recorte del dataset.