guptavasu1213/Yolo-Vehicle-Counter

🚗 Đếm xe thông minh với YOLOv3 — không đếm trùng dù xe trượt khung hình

Đếc xe trong video chính xác ngay cả khi YOLO bỏ sót một vài khung hình liên tiếp.

Yolo-Vehicle-Counter là một dự án Python dùng thuật toán YOLOv3 để thực hiện phát hiện và đếm các phương tiện giao thông trong video đầu vào. Dự án hỗ trợ 6 loại xe phổ biến: xe máy, xe buýt, ô tô, xe đạp, xe tải và tàu hỏa. Điểm đặc biệt là việc kết hợp theo dõi đối tượng (tracking) giúp tránh tình trạng đếm trùng lặp khi xe bị mất phát hiện tạm thời.

✨ Điểm nổi bật

• Phát hiện YOLOv3: Áp dụng mạng nơơn CNN thời gian thực để nhận diện vật thể trong từng khung hình video.
• Đếm 6 loại xe: Hỗ trợ phát hiện và đếm xe máy, xe buýt, ô tô, xe đạp, xe tải và tàu hỏa.
• Theo dõi không đếm lại: Khi xe đã được đếm một lần, thuật toán sẽ theo dõi chúng và không tính vào số đếm nếu xuất hiện lại trong các khung hình tiếp theo.
• Xử lý mất phát hiện tạm thời: Lưu trữ kết quả phát hiện trong 10 khung hình gần nhất để so sánh và xác định lại xe khi bị bỏ sót.
• Hỗ trợ CPU và GPU: Cấu hình chạy trên CPU mặc định, có thể bật GPU để tăng tốc độ xử lý.
• Tùy chỉnh độ tin cậy: Cho phép người dùng thiết lập ngưỡng confidence và threshold cho phép tùy theo nhu cầu.
• Xuất video kết quả: Ghi lại video đầu ra với hộp giới hạn và số lượng xe đã được đếm.

💡 Vì sao đáng chú ý

Khác với các dự án chỉ dừng lại ở việc đếm mỗi khi phát hiện được xe, dự án này giải quyết vấn đề mất phát hiện tạm thời — một thách thức phổ biến khi dùng YOLO trên video thực tế. Bằng cách lưu trữ thông tin các khung hình trước đó và so sánh khoảng cách trung tâm của các hộp giới hạn, thuật toán có thể phân biệt giữa xe mới và xe đã được đếm.

Phương pháp này đặc biệt hiệu quả trong môi trường đường cao tốc hoặc giao lộ đông xe, nơi các phương tiện thường xuất hiện, biến mất và xuất hiện lại do kích thước, tốc độ di chuyển hoặc góc nhìn. Dự án phù hợp cho các ứng dụng như giám sát giao thông, phân tích lư lượng xe qua lại trên các tuyến đường.

#YOLOv3 #VehicleDetection #ObjectTracking #Python #OpenCV #TiniX