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.