https://github.com/homuler/pyon2-detector

anime face detector and model learner with deep learning.
https://github.com/homuler/pyon2-detector

Last synced: about 2 months ago
JSON representation

anime face detector and model learner with deep learning.

• Host: GitHub
• URL: https://github.com/homuler/pyon2-detector
• Owner: homuler
• Created: 2015-09-21T15:24:52.000Z (over 9 years ago)
• Default Branch: master
• Last Pushed: 2015-09-22T02:50:51.000Z (over 9 years ago)
• Last Synced: 2024-03-23T13:00:29.837Z (10 months ago)
• Language: Python
• Size: 1.46 MB
• Stars: 15
• Watchers: 2
• Forks: 1
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# Pyon2-Detector

Scripts to train "Anime face detector" model using deep learning, and detect anime faces with the trained model.  

For more details, please refer to [this page](http://qiita.com/homulerdora/items/9a9af1481bf63470731a).

## Prepare Dataset

place dataset like this.

```

  dataset

   /train

     /0      -- 0 labeled dataset

     /1      -- 1 labeled dataset

     ...

   /valid

     /0      -- 0 labeled dataset

     /1      -- 1 labeled dataset

     ...

```

## Usage

```shell

# resize images

./pyon2-detector.py gendata --size=[output image size] --rawdata=[dataset directory] --output=[resized dataset directory]

# generate mean file

./pyon2-detector.py genmean --train=[training dataset directory] --output=[mean file path]

# generate train data list and validation data list

./pyon2-detector.py genlist --dataset=[training data directory] --output=[training data list]

./pyon2-detector.py genlist --dataset=[validation data directory] --output=[validation data list]

# train model

./pyon2-detector.py learn --train=[training data list] --valid=[validation data list] --gpu=[gpu flag] --mean=[mean file path] --model=[model name]

# detect faces

./pyon2-detector.py detect --gpu=[gpu flag] --mean=[mean file path] --model=[model dump file path] --input=[input image path] --output=[output image path]

```

## Reqirements

1. chainer >= 1.3.0

1. OpenCV >= 3.0