首页 > 人工智能 > 深度学习 > 基于Keras的三维人脸重建
这是一个非常炫酷的成就,能将人脸特征重建为三维空间的模型。
预训练模型最初是用torch实现的。
此代码是把torch实现的VRN模型转换为keras的模型。
依赖:
keras,custom_layers,h5py
1.下载预训练模型。
位置参见http://cs.nott.ac.uk/~psxasj/download.php?file=vrn-unguided.t7
2.使用样例:
输入是3x192x192的图像
参见Example-Usage.ipynb。
样例代码:
from keras.models import load_model
import custom_layers
custom_objects = {
'Conv': custom_layers.Conv,
'BatchNorm': custom_layers.BatchNorm,
'UpSamplingBilinear': custom_layers.UpSamplingBilinear
}
model = load_model('vrn-unguided-keras.h5', custom_objects=custom_objects)
3.转换结果的keras模型
位置参见https://drive.google.com/file/d/1oh8Zpe4wh00iXcm8ztRsi5ZL6GMkHdjj/view?usp=sharing
4.Tensorflow的模型
位置参见 https://drive.google.com/file/d/1THX-x6TR8Qg7zFfaFXU3cFd9PCZp22IY/view?usp=sharing
对于keras的学习,可参考如下书籍。
Keras图像深度学习实战
[tag]Keras,图像识别,深度学习,数挖掘,算法,
[content]一本不可多得的与图像处理紧密相连介绍的深度学习书籍,值得你花点时间阅读。
[link] https://u17749188.pipipan.com/fs/17749188-355855744
来自 “ ITPUB博客 ” ,链接:https://blog.itpub.net/69914286/viewspace-2639201/ ,如需转载,请注明出处,否则将追究法律责任。