从2D图像生成3D模型，基于Im2Avatar
原文连接：Im2Avatar: Colorful 3D Reconstruction from a Single Imagegithub.compython
这个模型不只能够生成3D模型，还能够生成颜色，如今我就介绍一下生成形状的部分，这个网络不是end to end的形式因此形状和颜色是分别训练的。源代码实际上是python2写的，稍微改一下就能够用在python3中。git
它的网络我大体列一下，也欢迎指正：github
编码：网络
解码：ui
最后输出的是64*64*64*1的h5文件编码
训练的时候若是内存不够用，能够把batch设置小一些。3d
源代码im2avatarblog
我写的可视化代码visualization
ip
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
这是个人各个库的版本，很重要。内存