Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
merge.
  • Loading branch information
yu4u committed Apr 13, 2017
2 parents 8c726b1 + 00e6b62 commit 454e4f3
Show file tree
Hide file tree
Showing 2 changed files with 117 additions and 1 deletion.
9 changes: 8 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -2,14 +2,15 @@
This is a Keras implementation of a CNN network for estimating age and gender from a face image.
In training, [the IMDB-WIKI dataset](https://data.vision.ee.ethz.ch/cvl/rrothe/imdb-wiki/) is used.


## Dependencies
- Python3.5+
- Keras
- scipy, numpy, Pandas, tqdm
- OpenCV3

## Usage

## Usage
Download the dataset. The dataset is downloaded and extracted to the `data` directory.

```sh
Expand All @@ -28,9 +29,15 @@ Train the network using the training data created above.
python3 train.py --input data/imdb_db.mat
```


## Network architecture
In [the original paper](https://www.vision.ee.ethz.ch/en/publications/papers/articles/eth_biwi_01299.pdf), the pretrained VGG network is adopted.
Here the Wide Residual Network (WideResNet) is trained from scratch.
I modified the @asmith26's implementation of the WideResNet; two classification layers (for age and gender estimation) are added on the top of the WideResNet.
Note that age and gender are estimated independently using different two CNNs.


## Results
Trained on imdb, tested on wiki.

![result](https://github.com/yu4u/age-gender-estimation/wiki/images/result.png)
109 changes: 109 additions & 0 deletions predict.ipynb

Large diffs are not rendered by default.

0 comments on commit 454e4f3

Please sign in to comment.