Skip to content

MXNET/Gluon Implementation of End-to-end 3D Face Reconstruction with Deep Neural Networks

License

Notifications You must be signed in to change notification settings

ShownX/mxnet-E2FAR

Folders and files

NameName
Last commit message
Last commit date

Latest commit

5974fa2 · Jan 25, 2018

History

5 Commits
Jan 25, 2018
Jan 25, 2018
Jan 25, 2018
Jan 25, 2018
Jan 25, 2018
Jan 25, 2018
Jan 25, 2018
Jan 25, 2018
Jan 25, 2018

Repository files navigation

mxnet-E2FAR

This is a MXNet/Gluon Implementation of End-to-end 3D Face Reconstruction with Deep Neural Networks.

  1. Download VGG-Face and convert it to the mxnet-weights by running the caffe_converter:
    python $MXNET/tools/caffe_converter/convert_model.py prototxt weights params_name
    

Put the weights into the folder ckpt/VGG-Face

  1. Prepare the dataset

  2. For train your dataset, you may need to change the dataset in the main code to fit your dataset

  3. Run the code:

    # fine-tune the branch and fully connected layers
    python E2FAR.py --pretrained --freeze --epoch 10
    
    # fine-tune whole network
    python E2FAR.py --start_epoch 10
    

If you use this code, pls mention this repo and cite the paper:

@InProceedings{Dou_2017_CVPR,
author = {Dou, Pengfei and Shah, Shishir K. and Kakadiaris, Ioannis A.},
title = {End-To-End 3D Face Reconstruction With Deep Neural Networks},
booktitle = {The IEEE Conference on Computer Vision and Pattern Recognition (CVPR)},
month = {July},
year = {2017}
}

Known issues

dataloader is very slow and cannot make fully usage of GPU training. You can use record io to pack the image and do more augmentation.

About

MXNET/Gluon Implementation of End-to-end 3D Face Reconstruction with Deep Neural Networks

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages