/AnimeFaceBoxes

Fast anime face detection

Primary LanguagePythonApache License 2.0Apache-2.0

AnimeFaceBoxes

Super fast anime face detection (> 200 fps for 512x512 resolution, RTX2080 + i7 9700k)

Dependencies

  • Python 3.6+ (Anaconda)
  • PyTorch-1.0 +
  • OpenCV3 (Python)

Usage

  • build nms: sh make.sh
  • Manual data labeling: LabelFaceBox.py (you can skip this if you have danbooru2018 dataset)
  • labeled data: faceboxes (600+ labels for danbooru2018 dataset)
  • train: MyTrain.py
  • eval: eval.py

alt text

codes borrowed a lot from FaceBoxes.PyTorch