vehicle-classification-cnn

🚗 Vehicle Classification using Multiple CNN Models

Repo này cung cấp nhiều mô hình CNN đã huấn luyện để phân loại ảnh phương tiện giao thông, phục vụ nghiên cứu, học tập và so sánh kiến trúc Deep Learning.

---

📌 Bài toán

Phân loại ảnh phương tiện giao thông thành 5 lớp:

• xe_dap
• xe_bus
• xe_may
• o_to
• xe_tai

---

🧠 Các mô hình đã huấn luyện

---

⚙️ Công nghệ sử dụng

• Python 3
• TensorFlow / Keras
• NumPy
• Hugging Face Hub

---

📂 Cấu trúc repo

vehicle-classification-cnn/ ├── best_MobileNetV2.h5 ├── best_ResNet50.h5 ├── best_VGG16.h5 ├── best_EfficientNetB0.h5 ├── best_InceptionV3.h5 ├── labels.json └── README.md

---

🏷️ Nhãn phân loại

labels.json chứa mapping: { "Xe đạp" : 0 , "Xe buýt" : 1 , "Ô tô" : 2 , "Xe máy" : 3 , "Xe tải" : 4 }

---

🚀 Cách load model (ví dụ)

from huggingface_hub import hf_hub_download from tensorflow.keras.models import load_model

model_path = hf_hub_download( repo_id="Annga121/vehicle-classification-cnn", filename="best_EfficientNetB0.h5" )

model = load_model(model_path)

---

👤 Tác giả

Hugging Face: Annga121