Skip to content

Commit

Permalink
Update age model, Add benchmark result
Browse files Browse the repository at this point in the history
  • Loading branch information
abars committed Jan 6, 2019
1 parent 7ae97c8 commit 5b833c8
Show file tree
Hide file tree
Showing 5 changed files with 13 additions and 9 deletions.
16 changes: 10 additions & 6 deletions README.md
Expand Up @@ -103,7 +103,7 @@ This is a convert script.

This is a converted model.

<http://www.abars.biz/keras/yolov2_tiny-face.h5>
<https://github.com/abars/YoloKerasFaceDetection/releases/download/1.10/yolov2_tiny-face.h5>

## Age and Gender classification

Expand Down Expand Up @@ -171,17 +171,21 @@ Test classifier task using keras.

### Training result

Age101 (IMDB)
Age101 (IMDB) (EPOCHS=100)

<img src="https://github.com/abars/YoloKerasFaceDetection/blob/master/pretrain/agegender_age101_squeezenet.png" width="50%" height="50%">
<img src="https://github.com/abars/YoloKerasFaceDetection/blob/master/pretrain/agegender_age101_squeezenet_imdb.png" width="50%" height="50%">

<http://www.abars.biz/keras/agegender_age101_squeezenet.hdf5>
<img src="https://github.com/abars/YoloKerasFaceDetection/blob/master/pretrain/benchmark_age101_squeezenet_imdb.png" width="50%" height="50%">

Gender (IMDB)
<https://github.com/abars/YoloKerasFaceDetection/releases/download/1.10/agegender_age101_squeezenet_imdb.hdf5>

Gender (IMDB) (EPOCHS=25)

<img src="https://github.com/abars/YoloKerasFaceDetection/blob/master/pretrain/agegender_gender_squeezenet.png" width="50%" height="50%">

<http://www.abars.biz/keras/agegender_gender_squeezenet.hdf5>
<img src="https://github.com/abars/YoloKerasFaceDetection/blob/master/pretrain/benchmark_gender_squeezenet_imdb.png" width="50%" height="50%">

<https://github.com/abars/YoloKerasFaceDetection/releases/download/1.10/agegender_gender_squeezenet_imdb.hdf5>

# Related Work

Expand Down
6 changes: 3 additions & 3 deletions download_model.py
Expand Up @@ -13,23 +13,23 @@ def main(argv):
os.mkdir(OUTPUT_PATH)
print("1/3");
with open(OUTPUT_PATH+'agegender_age101_squeezenet.hdf5','wb') as f:
path="https://github.com/abars/YoloKerasFaceDetection/releases/download/1.0.0/agegender_age101_squeezenet.hdf5"
path="https://github.com/abars/YoloKerasFaceDetection/releases/download/1.10/agegender_age101_squeezenet_imdb.hdf5"
if sys.version_info >= (3,0):
f.write(request.urlopen(path).read())
else:
f.write(urllib2.urlopen(path).read())
f.close()
print("2/3");
with open(OUTPUT_PATH+'agegender_gender_squeezenet.hdf5','wb') as f:
path="https://github.com/abars/YoloKerasFaceDetection/releases/download/1.0.0/agegender_gender_squeezenet.hdf5"
path="https://github.com/abars/YoloKerasFaceDetection/releases/download/1.10/agegender_gender_squeezenet_imdb.hdf5"
if sys.version_info >= (3,0):
f.write(request.urlopen(path).read())
else:
f.write(urllib2.urlopen(path).read())
f.close()
print("3/3");
with open(OUTPUT_PATH+'yolov2_tiny-face.h5','wb') as f:
path="https://github.com/abars/YoloKerasFaceDetection/releases/download/1.0.0/yolov2_tiny-face.h5"
path="https://github.com/abars/YoloKerasFaceDetection/releases/download/1.10/yolov2_tiny-face.h5"
if sys.version_info >= (3,0):
f.write(request.urlopen(path).read())
else:
Expand Down
Binary file added pretrain/agegender_age101_squeezenet_imdb.png
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added pretrain/benchmark_age101_squeezenet_imdb.png
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added pretrain/benchmark_gender_squeezenet_imdb.png
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.

0 comments on commit 5b833c8

Please sign in to comment.