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.