从2D图像生成3D模型，基于Im2Avatar

原文链接：Im2Avatar: Colorful 3D Reconstruction from a Single Imagegithub.com

这个模型不仅可以生成3D模型，还可以生成颜色，现在我就介绍一下生成形状的部分，这个网络不是end to end的形式所以形状和颜色是分别训练的。源代码其实是python2写的，稍微改一下就可以用在python3中。

它的网络我大致列一下，也欢迎指正：

编码：

解码：

最后输出的是64*64*64*1的h5文件

训练的时候如果内存不够用，可以把batch设置小一些。

源代码im2avatar

我写的可视化代码visualization 

python 3.6.0

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

这是我的各个库的版本，很重要。