Ireland
kaylode/vehicle-counting
Vehicle counting using Pytorch
🚗 Truy vết & đếm phương tiện thông minh với YOLOv5 + DeepSORT
Kết hợp phát hiện đối tượng mạnh mẽ và truy vết liên tưởng chất lượng cao, dự án mang lại giải pháp đếm xe hiệu quả trên video giao thông.
Dự án vehicle-counting của kaylode sử dụng YOLOv5 để phát hiện phương tiện và DeepSORT để truy vết chúng trong video. Chỉ với một file video đầu vào, hệ thống sẽ tự động đếm số lượng từng loại phương tiện theo từng hướng di chuyển, đồng thời xuất kết quả ra file CSV chi tiết theo từng khung hình.
✨ Điểm nổi bật
- Phát hiện bằng YOLOv5: Chỉ xét các đối tượng nằm trong vùng quan tâm (ROI).
- Truy vết bằng DeepSORT: Không cần huấn luyện lại, chỉ cần dùng mô hình đã có sẵn.
- Gán hướng di chuyển: Dùng Cosine Similarity để gán track của đối tượng vào hướng tương ứng.
- Đếm theo hướng: Đếm từng loại xe trên từng hướng một cách chính xác.
- Xuất kết quả CSV: Bao gồm track_id, frame_id, box, màu sắc, nhãn, hướng, tọa độ đầu/cuối và khung hình xuất hiện.
- Hỗ trợ notebook Colab: Dễ dàng chạy thử ngay trên trình duyệt.
- Mô hình đã huấn luyện: Cung cấp trọng số YOLOv5s và YOLOv5m tinh chỉnh trên bộ dữ liệu AIC-HCMC-2020.
💡 Vì sao đáng chú ý
Không giống các giải pháp chỉ đếm xe đơn thuần, dự án này kết hợp DeepSORT giúp truy vết ổ định ID của từng phương tiện qua các khung hình, tránh đếm trùng lặp. Ngoài ra, việc phân tích hướng di chuyển thông qua Cosine Similarity giúp hỗ trợ các bài toán như phân tích lưu lượng giao thông theo hướng, rất hữu ích cho hệ thống giám sát đường bộ thông minh.
#VehicleCounting #YOLOv5 #DeepSORT #ObjectTracking #TiniX
Similar Projects
Discover other similar projects and AI models.
Ultralytics YOLO27, YOLO26, YOLO11, YOLOv8 — object detection, instance segmentation, semantic segmentation, image classification, pose estimation, object tracking
Ultralytics YOLOv5 in PyTorch for object detection, instance segmentation, classification, training, and export.
Implement a ChatGPT-like LLM in PyTorch from scratch, step by step
🧑🏫 60+ Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback, vit, ...), optimizers (adam, adabelief, sophia, ...), gans(cyclegan, stylegan2, ...), 🎮 reinforcement learning (ppo, dqn), capsnet, distillation, ... 🧠
🧑🏫 60+ Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback, vit, ...), optimizers (adam, adabelief, sophia, ...), gans(cyclegan, stylegan2, ...), 🎮 reinforcement learning (ppo, dqn), capsnet, distillation, ... 🧠
DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.