pollux435/mobilenet-v4-emotion

MobileNetV4 Emotion Recognition Model 🎭

FER2013 데이터셋으로 학습된 MobileNetV4 기반의 감정 인식 모델입니다.

모델 개요

이 모델은 MobileNetV4를 백본으로 사용하여 7가지 감정을 분류합니다:

• 😠 Angry (화남)
• 🤢 Disgust (혐오)
• 😨 Fear (두려움)
• 😊 Happy (행복)
• 😢 Sad (슬픔)
• 😲 Surprise (놀람)
• 😐 Neutral (무표정)

모델 아키텍처

• 백본: MobileNetV4 (timm/mobilenetv4_conv_medium.e500_r224_in1k)
• 분류기: 3층 완전연결층 (512 → 128 → 7)
• 정규화: Batch Normalization + Dropout (0.5)
• 입력 크기: 224×224 RGB 이미지

성능

• 검증 정확도(test): 62.50% (FER2013 기준)
• 모델 크기: 약 35MB
• 추론 속도: batch=256에서 1.41it/s

사용 방법

빠른 시작

from transformers import AutoConfig
from modeling_mobilenet_v4_emotion import MobileNetV4EmotionClassifier
from preprocessing import preprocess_image
import torch

# 모델 로드
config = AutoConfig.from_pretrained("pollux435/mobilenet-v4-emotion", trust_remote_code=True)
model = MobileNetV4EmotionClassifier.from_pretrained("pollux435/mobilenet-v4-emotion", trust_remote_code=True)
model.eval()

# 이미지 예측
image_path = "your_image.jpg"
pixel_values = preprocess_image(image_path)

with torch.no_grad():
    outputs = model(pixel_values)
    probabilities = torch.nn.functional.softmax(outputs.logits, dim=-1)
    
predicted_class_id = outputs.logits.argmax().item()
predicted_emotion = config.id2label[str(predicted_class_id)]
confidence = probabilities[0][predicted_class_id].item()

print(f"예측된 감정: {predicted_emotion}")
print(f"신뢰도: {confidence:.4f}")

전이학습

이 모델을 사용해서 쉽게 전이학습을 할 수 있습니다:

from transformers import AutoConfig
from modeling_mobilenet_v4_emotion import MobileNetV4EmotionClassifier

# 기존 모델 로드
model = MobileNetV4EmotionClassifier.from_pretrained("pollux435/mobilenet-v4-emotion", trust_remote_code=True)

# 새로운 클래스 수로 분류기 교체 (예: 3개 클래스)
import torch.nn as nn
in_features = model.classifier[-1].in_features
model.classifier[-1] = nn.Linear(in_features, 3)

# 이제 여러분의 데이터로 학습하세요!

설치 요구사항

pip install torch torchvision transformers timm pillow numpy

학습 정보

• 데이터셋: FER2013
• 학습 방법: 2단계 학습
  1. 분류기만 학습 (10 에포크)
  2. 전체 네트워크 미세조정 (15 에포크)
• 최적화: Adam 옵티마이저
• 학습률:
  • 분류기: 1e-3
  • 백본: 1e-5 (미세조정 시)

라벨 매핑

{
  "0": "angry",
  "1": "disgust", 
  "2": "fear",
  "3": "happy",
  "4": "sad",
  "5": "surprise",
  "6": "neutral"
}

예제 결과

입력 이미지	예측 감정	신뢰도
😊	Happy	0.89
😢	Sad	0.76
😠	Angry	0.82

인용

이 모델을 사용하실 경우 다음과 같이 인용해 주세요:

@misc{mobilenetv4-emotion-2024,
  title={MobileNetV4 Based Emotion Recognition Model},
  author={pollux435},
  year={2024},
  publisher={Hugging Face},
  note={Trained on FER2013 dataset}
}

라이선스

MIT License