AI一键将图片生成3D模型【TripoSR整合包】

1710568241663

介绍

TripoSR是stability.ai发布的支持从单个图像快速生成3D对象的AI大模型程序，这种新的图像到 3D 模型旨在满足娱乐、游戏、工业设计和建筑专业人士日益增长的需求，并提供用于可视化详细 3D 对象的响应式输出。

1710519072131

TripoSR可以在其他模型的一小部分时间内创建详细的3D模型。在 Nvidia A100 上测试时，它可以在大约 0.5 秒内生成草稿质量的 3D 输出（纹理网格），优于其他开放图像到 3D 模型，例如 OpenLRM。除了速度之外，无论是否使用 GPU，我们的模型都可以完全访问。

1710519109412

img

项目地址：https://github.com/VAST-AI-Research/TripoSR

官网：https://stability.ai/news/triposr-3d-generation

在线示例：https://huggingface.co/spaces/stabilityai/TripoSR

夸克云盘：https://pan.quark.cn/s/d720b89a5358

123云盘：https://www.123pan.com/s/9T6A-OS9xd.html

整合包启动

WIN:双击run.bat即可，会自动加载下载模型

MAC:

安装依赖

1
pip install -r requirements.txt

本地 Gradio 应用程序

1
python gradio_app.py

手动推理

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

这会将重建的 3D 模型保存到 。您还可以指定多个以空格分隔的图像路径。默认选项的单个图像输入大约需要 6GB VRAM。output/

有关此脚本的详细用法，请使用 python run.py --help

源码环境要求

  • Python >= 3.8
  • 安装 CUDA（如果可用）
  • 根据您的平台安装 PyTorch： https://pytorch.org/get-started/locally/ [请确保本地安装的 CUDA 主要版本与 PyTorch 附带的 CUDA 主要版本匹配。例如，如果您安装了 CUDA 11.x，请确保安装使用 CUDA 11.x 编译的 PyTorch。
  • 更新 setuptoolspip install --upgrade setuptools
  • 通过以下方式安装其他依赖项pip install -r requirements.txt