hysts/anime-face-detector

Anime Face Detector using mmdet and mmpose

What it solves

이미지에서 애니메이션 스타일의 얼굴을 자동으로 감지하고 얼굴 랜드마크를 추정하는 방법을 제공합니다. 이는 일반적인 인간 얼굴 감지기에서는 어려운 경우가 많습니다.

How it works

이 프로젝트는 PyTorch를 사용하여 사전 학습된 모델을 실행해 감지와 랜드마크 추정을 수행합니다. Faster R-CNN과 YOLOv3 중 하나를 선택하여 정면에 가까운 애니메이션 얼굴의 경계 상자를 감지하고, HRNetV2 모델을 사용해 얼굴에 28개의 특정 랜드마크 포인트를 예측합니다.

Who it’s for

애니메이션 스타일 이미지 작업을 하는 개발자와 연구자를 대상으로 하며, 얼굴을 찾고 주요 얼굴 특징을 식별하여 추가 처리를 해야 하는 경우에 적합합니다.

Highlights

  • Multiple Detection Options: Supports both Faster R-CNN and YOLOv3 architectures.
  • Landmark Estimation: Predicts 28 distinct facial landmark points.
  • Simplified Dependencies: Runs in plain PyTorch without requiring the OpenMMLab runtime.
  • Easy Integration: Provides a simple Python API and automatic weight downloading via Hugging Face Hub.