单图像生成3D动画模型TripoSR的部署过程
文章目录
- torchmcubes安装问题
- 问题1:CUDA安装(只有runtime)
- 问题2:没有NVTX
- 手动下载模型及相关配置修改
- 下载U2Net
- OPENGL依赖处理
- 940MX上的结果
- 3060上的结果
- GPU修改
- 解决方案1:指定为CPU
- 解决方案2:tensor放到cuda上
- obj与glb的区别
最近想做一个3D建模的事情,在做了一番技术调研之后,发现现在的一个技术路径是通过单张的图像直接生成对应的3D模型,而其中的一个模型:TripoSR的应用还比较多。决定深入研究一下,和之前不一样,这次打算在读论文之前,把demo先跑起来。
把模型跑起来还是有点小坑的。
- 项目的GITHUB地址:TripoSR
- 论文下载地址:论文下载
torchmcubes安装问题
把代码从GITHUB上下载下来之后,第一步自然就是安装python的依赖包:
pip3 install -r requirement.txt
其他的基本上都是顺利一把安装完成,但是里面的torchmcubes项目我碰到点问题。
git+https://github.com/tatsy/torchmcubes.git
torchmcubes 是一个基于 PyTorch 的 Python 库,用于从三维体素数据或隐式函数中提取表面网格(surface mesh)。它实现了 Marching Cubes 算法ÿ