hfuwcs213/birds_detect
0
Ứng dụng Phát hiện và Phân loại Loài Chim
Đây là một ứng dụng web đơn giản sử dụng mô hình YOLO để phát hiện đối tượng (chim) và mô hình EfficientNet để phân loại loài chim từ ảnh tải lên.
Cách sử dụng:
- Tải ảnh lên bằng form bên dưới.
- Ứng dụng sẽ chạy mô hình YOLO để tìm kiếm các đối tượng.
- Đối với mỗi đối tượng được YOLO phát hiện, ứng dụng sẽ cắt vùng ảnh đó và đưa vào mô hình EfficientNet để phân loại loài chim.
- Kết quả (ảnh với bounding box và tên loài/confidence) sẽ được hiển thị cùng với danh sách chi tiết.
Công nghệ sử dụng:
- Frontend: HTML/CSS/Flask Templates
- Backend: Python Flask
- Object Detection: YOLO11m (từ thư viện ultralytics)
- Bird Species Classification: EfficientNet-B3 (PyTorch)
- Web Server: Gunicorn
- Deployment: Docker trên Hugging Face Spaces
Lưu ý:
- Quá trình xử lý ảnh có thể mất một chút thời gian tùy thuộc vào kích thước ảnh và tài nguyên của Space.
- Phiên bản demo này chạy trên CPU (
instance_type: cpu-basic). Nên chậm 👍.
Liên hệ: [hehe]
