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

别来无恙 提交于 2019-12-06 21:25:24

原文链接: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


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


标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!