Anime head detection

源碼地址：

https://github.com/jiuxianghedonglu/AnimeHeadDetection

 

  

1. Dataset

All the images in the dataset are collected from konachan.net and labeled by myself with LabelImg. Download dataset

2. Usage

2.1 Requirements

• python 3.6
• pytorch 1.3.0
• torchvision 0.4.1
• opencv-python 3.4.1

2.2 Training

1. Download dataset. Following are dataset directory trees and labeled file format.

    data
        imgs
            anime_0001.jpg
            anime_0002.jpg
            ...
        tr_label.csv
        val_label.csv
        ...