Codigo que permite realizar el conteo y clasificacion de vehiculos con Yolov4.
Este repositorio utiliza conteo por linea virtual y conteo por regiones. El cual, perimite identificar cuantos y que tipo de vehiculos de desplazan de una region x a una region y.
El modelo fue entrenado usando Google colab. Fue usando un tamano del lote de 64 con una taza de aprendizaje de 0.003.
Este programa esta dividido en cuatro etapas fundamentales:
- Deteccion
- Identificacion
- Trayectoria
- Conteo
Ademas, consta con el uso de otras funciones:
- Seguimiento
- Re-identificacion
- conteo por regiones
Fueron utilizadas las metricas de pascal para evaluar el modelo de deteccion.
Modelo | mAP (%) | IoU (%) |
---|---|---|
YOLOv2 | 60 | 54 |
YOLOv3 | 64 | 58 |
YOLOv4 | 69 | 63 |
Instals estos requerimientos
pip install -r requirements.txt