Dự án này là một dự án toàn diện về thị giác máy tính và học máy trong bóng đá. Dự án sử dụng YOLO để phát hiện đối tượng, Kmeans để phân đoạn pixel, optical flow để theo dõi chuyển động và phép biến đổi phối cảnh để phân tích chuyển động của cầu thủ trong các video bóng đá.
-
Sử dụng model đã được train qua YOLOv5
-
Có thể tùy chỉnh để train model qua các dataset khác nhau
XEM THÊM ==> Hướng dẫn cài đặt chi tiết
Nguồn: Sharecode.vn
Tải các thư viện requirements về
pip install -r requirements.txt
Chạy file main.py
[Lưu ý]
- Trước khi dùng cần xóa file ở folder stubs để xóa các file pkl đã được phân tích trước đó, sau đó chạy hàm main để phân tích file pkl mới
- Model trong file tải là model đã được train sẵn qua dataset (có ghi trong README)
- Thay thế file trong input_videos và đổi tên file ở hàm main nếu muốn test các video khác.