☰
TripoSR 单图转 3D 模型上手指南:0.5 秒出一张网格,6GB 显存就够
2026/10/16 5:56:57 网站建设 项目流程

TripoSR 单图转 3D 模型上手指南:0.5 秒出一张网格,6GB 显存就够

【免费下载链接】TripoSRTripoSR: Fast 3D Object Reconstruction from a Single Image项目地址: https://gitcode.com/GitHub_Trending/tr/TripoSR

TripoSR 是 Tripo AI 与 Stability AI 联合开源的前馈式 3D 重建模型:一张普通照片进,0.5 秒内出一张可下载的 3D 网格,单图约 6GB 显存。它不做迭代优化,一次前向推理直接从图像得到几何,适合把手机实拍快速变成建模、打印或引擎可用的资产。

先花 1 分钟确认它适不适合你

先说结论:它解决的是"一张照片变一张可用网格",不是高精度多视图扫描。

  • 适合谁:手里有一张物件、产品或场景照片,想要 OBJ / GLB 文件去建模、打印或进引擎,就是它的场景。
  • 不适合谁:需要多视角精密对位、或大面积遮挡的复杂场景。背面与遮挡处的细节是这类单图方法的共性短板,别期待过高。
  • 环境门槛:Python ≥ 3.8;有 GPU 时单图约 6GB 显存;NVIDIA 显卡的话,本地 CUDA 主版本必须与 PyTorch 内置版本一致(本地 CUDA 11.x 就装 CUDA 11.x 编译的 PyTorch)。
  • 产出物:默认是带顶点色的网格文件(mesh.obj 或 mesh.glb);开启纹理烘焙后,多一张 2D 纹理贴图。

最短路径跑起来

先说结论:一条命令链就能从克隆到出网格,全程不需要手动抠图(run.py 内置 rembg 自动去背景和前景归一化)。

先克隆仓库并装依赖,PyTorch 本身请按你的平台提前装好:

git clone https://gitcode.com/GitHub_Trending/tr/TripoSR cd TripoSR pip install --upgrade setuptools pip install -r requirements.txt

注意:setuptools 必须先到 49.6.0 以上,否则 requirements 里的 torchmcubes 源码编译容易失败。⏱️ 首次运行还会下载预训练权重,第一次启动比后续慢。

再跑仓库自带的示例图:

python run.py examples/chair.png --output-dir output/

你会看到终端依次打印模型加载、图像处理、网格提取的耗时;结束后output/0/里多出一个mesh.obj和预处理后的input.png,拿任意 3D 软件打开就是重建结果。

按你的目标选一条路

默认命令出的是顶点色网格,下面按目的挑一条加料路线。

要 UV 贴图进渲染管线:加 --bake-texture

顶点色精度有限,需要贴图时加--bake-texture,纹理图集大小用--texture-resolution指定:

python run.py examples/teapot.png --bake-texture --texture-resolution 2048

得到output/0/下带 UV 的mesh.obj和一张 2048×2048 的texture.png,烘焙逻辑在 tsr/bake_texture.py。

想用界面调参:启动 Gradio

不想敲命令就起本地网页:

python gradio_app.py

浏览器打开默认端口 7860 上传图片即可,页面提供 Foreground Ratio(默认 0.85)和 Marching Cubes Resolution(默认 256)两个滑块,结果可直接下载 OBJ / GLB。注意:页面 3D 预览存在翻转/偏色显示,下载的文件才是正确结果;对界面自带的示例图请关闭 "Remove Background" 再运行。

要批量处理或旋转预览

多张图按空格分隔一次传给run.py即可逐张处理;想要旋转视频再加--render,输出目录会多出 30 视角的render.mp4。

用到再看的细节

🔧 只在你要改变默认行为时才需要动这些参数,其余保持默认即可:

参数默认值什么时候需要改
--mc-resolution256想要更密的网格细节,愿意用时间换精度时
--texture-resolution2048仅--bake-texture生效,调纹理图集大小
--chunk-size8192显存紧张时调小省显存,换更多计算时间
--foreground-ratio0.85重建位置或大小不对时的第一调参项
--no-remove-bg关输入已是灰底居中图时跳过自动抠图
--model-save-formatobj想直接导出 glb 时改成 glb
--render关想要 NeRF 渲染的 30 视角旋转视频时
--output-diroutput/想把结果放到别的目录时

卡住了?按顺序排查

🔍 卡住的大多集中在这三类,按顺序对号入座:

问题 1:torchmcubes 报 CUDA 相关错误

  • 现象:启动时报AttributeError: module 'torchmcubes_module' has no attribute 'mcubes_cuda',或提示 torchmcubes was not compiled with CUDA support。
  • 原因:本地 CUDA 主版本与 PyTorch 内置版本不一致,或 setuptools 低于 49.6.0,扩展在无 CUDA 环境下被编译。
  • 修复:先核对 CUDA 版本与 setuptools 版本,然后重装该依赖:
pip uninstall torchmcubes && pip install -r requirements.txt

问题 2:显存不够,CUDA out of memory

  • 现象:批量图或大分辨率输入时进程被 OOM 杀掉或抛 CUDA 错误。
  • 原因:默认单图约 6GB 显存已超预算,且默认--chunk-size8192 的表面提取批次偏大。
  • 修复:把批次调小重跑(越小越省显存,耗时越长):
python run.py examples/chair.png --chunk-size 2048

问题 3:网格被切掉或主体偏位

  • 现象:重建的网格不完整,或主体在画布中位置、大小不对。
  • 原因:前景占比与画面中真实主体比例不匹配,或主体没有居中。
  • 修复:重跑同一条命令并加上--foreground-ratio 0.9(Gradio 页面里对应 Foreground Ratio 滑块,直接拖动即可)。

先跑通最小命令,再按需求叠加--bake-texture、批量路径与--render,效果不理想时优先动--foreground-ratio而不是分辨率。项目整体为 MIT 许可证,源码、预训练权重均可直接用于研究与商业场景,落地没有授权顾虑。核心重建逻辑在 tsr/system.py,想深入时从这里读起。

【免费下载链接】TripoSRTripoSR: Fast 3D Object Reconstruction from a Single Image项目地址: https://gitcode.com/GitHub_Trending/tr/TripoSR

创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

需要专业的网站建设服务?

联系我们获取免费的网站建设咨询和方案报价,让我们帮助您实现业务目标

立即咨询