基於Keras的三維人臉重建

renwoxing2019發表於2019-03-24

這是一個非常炫酷的成就,能將人臉特徵重建為三維空間的模型。

預訓練模型最初是用torch實現的。

此程式碼是把torch實現的VRN模型轉換為keras的模型。

依賴:

keras,custom_layers,h5py

1.下載預訓練模型。

位置參見~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模型

位置參見

4.Tensorflow的模型

位置參見


對於keras的學習,可參考如下書籍。

Keras影像深度學習實戰
[tag]Keras,影像識別,深度學習,數挖掘,演算法,
[content]一本不可多得的與影像處理緊密相連介紹的深度學習書籍,值得你花點時間閱讀。

 [link]

 


來自 “ ITPUB部落格 ” ,連結:http://blog.itpub.net/69914286/viewspace-2639201/,如需轉載,請註明出處,否則將追究法律責任。

相關文章