AnimeHeads / README.md
nyuuzyou's picture
Update README.md
7498d55
metadata
license: wtfpl
tags:
  - object-detection
datasets:
  - nyuuzyou/AnimeHeadsv3

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:

    Example image 1 Example image 2 Example image 3

  • 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:

    Example image 1 Example image 2 Example image 3

  • 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:

    Example image 1 Example image 2 Example image 3

  • Model file: animeheads.pt