Skip to content

WynMew/AnimeFaceRanker

Repository files navigation

996.icu

AnimeFaceRanker

Check detected Anime Face Quality (delete noise data). Implementated with HRNet

Dependencies

  • Python 3.6+ (Anaconda)
  • PyTorch-0.4.0 +
  • scipy, numpy, sklearn etc.
  • OpenCV3 (Python)

Usage

  • Download Danbooru dataset(or any other dataset you want to process)

  • face detection with lbpcascade_animeface or danbooru-utility

  • Make your training dataset by hand (Good face detection and bad case, a few thousand each). You may skip this.

  • Training with FaceClassifier.py. You may skip this.

  • Rank your face detection with FaceRanker.py

  • My trained model. 提取码: hjpn; Eatraction code: hjpn

Good Face Samples:

  • alt text

Bad Case:

  • alt text