Um sistema inteligente de contagem de veículos com análise de semáforo, detectando infrações e evitando contagens duplicadas usando visão computacional.
- Contagem de veículos baseada em uma linha virtual.
- Análise do estado do semáforo (vermelho, amarelo, verde) via espaço de cor HSV.
- Rastreamento contínuo de veículos usando YOLO e ByteTrack.
- Detecção de infrações quando veículos cruzam a linha durante o sinal vermelho.
- Evita duplicações com IDs únicos para cada veículo.
-
Clone o repositório:
git clone https://github.com/seu-usuario/Smart-Traffic-Vehicle-Counter.git cd Smart-Traffic-Vehicle-Counter -
Instale as dependências:
pip install -r requirements.txt
-
Baixe o modelo pré-treinado YOLO (ex:
volo11s.pt) e coloque-o na pastamodels/.
- Execute o notebook Jupyter:
jupyter notebook notebooks/Smart_Traffic_Analysis.ipynb
Parâmetros:
video: Caminho do vídeo de entrada.model: Caminho do modelo YOLO.line_y: Posição Y da linha virtual de contagem (padrão: 600).
Distribuído sob a licença MIT. Veja LICENSE para mais detalhes.
- Modelo YOLO: Ultralytics
- Rastreamento: ByteTrack