Anime Head Detection Models

This repository contains three object detection models based on YOLOv5l and YOLOv8l architectures. The images in the dataset were taken from anime and art, with manga pages added to AnimeHeadsv3.

Versions

AnimeHeadsv3

• Description: The third and last version of the project based on YOLOv8l. The training set contains 8037 images.

• Augmentation parameters:

  • Outputs per training example: 3
  • Flip: Horizontal
  • Saturation: Between -40% and +40%
  • Blur: Up to 4px
  • Noise: Up to 4% of pixels

• Example images:

  

• Model file: animeheadsv3.pt

AnimeHeadsv2

• Description: The second version of the project based on YOLOv8l. The training set contains 4632 images.

• Augmentation parameters:

  • Outputs per training example: 3
  • Blur: Up to 2.75px
  • Noise: Up to 5% of pixels

• Example images:

  

• Model file: animeheadsv2.pt

AnimeHeads

• Description: The first version of the project based on YOLOv5l. The training set contains 886 images.

• Augmentation parameters: None

• Example images:

  

• Model file: animeheads.pt