DucTai368/AI_sketch-recognition-v2

AI Sketch Recognition V2

Đây là mô hình nhận dạng nét vẽ (Sketch Recognition) cho đồ án tốt nghiệp tại Đại học Duy Tân (DTU).

🚀 Thông số kỹ thuật (Specifications)

• Kiến trúc: Multi-Scale CNN + Multi-Head Attention + Bi-LSTM (CRNN V2).
• Dữ liệu: 5.1 triệu mẫu từ tập dữ liệu Google QuickDraw (15,000 mẫu/lớp).
• Đầu vào (Input): Chuỗi tọa độ Delta (128, 3) bao gồm [dx, dy, pen_state].
• Số lớp phân loại: 345 lớp từ vựng tiếng Anh.

📊 Hiệu năng (Performance)

• Top-1 Accuracy: ~85% (Trained on NVIDIA A100).
• Top-5 Accuracy: ~95% (Phù hợp cho ứng dụng giáo dục trẻ em).

🛠️ Cách sử dụng (Usage)

from huggingface_hub import hf_hub_download
import tensorflow as tf

model_path = hf_hub_download(repo_id="DucTai368/AI_sketch-recognition-v2", filename="best_model_v2.keras")
model = tf.keras.models.load_model(model_path, compile=False)