Hướng dẫn training custom model YOLO v8 để đếm xe trên đường cao tốc

Trong bài viết trước, chúng ta đã tìm hiểu cách phát hiện xe trên đường cao tốc bằng Yolov8. Tuy nhiên, việc phát hiện xe chỉ là bước khởi đầu, để có thể quản lý giao thông hiệu quả, điều quan trọng là phải đếm chính xác số lượng xe di chuyển qua các điểm quan sát. Bài viết này sẽ hướng dẫn bạn cách đếm xe trên đường cao tốc.

Clone repository

Đầu tiên cần cài đặt môi trường và huấn luyện mô hình Yolov8 để phát hiện xe. Để biết chi tiết cách làm, bạn có thể tham khảo bài viết Phát hiện xe trên đường cao tốc bằng Yolov8
Download source code cuối bài viết

Đếm số lượng xe

Sau khi xe được phát hiện, sẽ tiến hành theo dõi (tracking) và đếm số lượng xe (counting), trong ultralytics đã có sẵn module tracker và counter. Có hai chế độ đếm xe mà bạn có thể lựa chọn:

    Đếm xe khi chúng vượt qua một vạch ngang. Mở cmd và sử dụng lệnh
    python .\count.py --model .\runs\detect\train\weights\best.pt --count_mode line

    Đếm xe khi chúng đi qua một vùng được xác định trước. Mở cmd và sử dụng lệnh
    python .\count.py --model .\runs\detect\train\weights\best.pt --count_mode region

Download

Repo Github: https://github.com/thigiacmaytinh/HighwayCounter_YOLOv8

Leave a Reply