原文链接:Im2Avatar: Colorful 3D Reconstruction from a Single Imagegithub.com
这个模型不仅可以生成3D模型,还可以生成颜色,现在我就介绍一下生成形状的部分,这个网络不是end to end的形式所以形状和颜色是分别训练的。源代码其实是python2写的,稍微改一下就可以用在python3中。
它的网络我大致列一下,也欢迎指正:
编码:
解码:
最后输出的是64*64*64*1的h5文件
训练的时候如果内存不够用,可以把batch设置小一些。
源代码im2avatar
我写的可视化代码visualization
h5py 2.8.0
mayavi 4.5.0+vtk71
numpy 1.14.5+mkl
opencv-python 3.4.1.15
PyQt4 4.11.4
scipy 1.1.0
traits 4.6.0
traitsui 6.0.0
VTK 7.1.1
这是我的各个库的版本,很重要。