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