AI一键将图片生成3D模型【TripoSR整合包】
介绍
TripoSR是stability.ai发布的支持从单个图像快速生成3D对象的AI大模型程序,这种新的图像到 3D 模型旨在满足娱乐、游戏、工业设计和建筑专业人士日益增长的需求,并提供用于可视化详细 3D 对象的响应式输出。
TripoSR可以在其他模型的一小部分时间内创建详细的3D模型。在 Nvidia A100 上测试时,它可以在大约 0.5 秒内生成草稿质量的 3D 输出(纹理网格),优于其他开放图像到 3D 模型,例如 OpenLRM。除了速度之外,无论是否使用 GPU,我们的模型都可以完全访问。
项目地址: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:
安装依赖
pip install -r requirements.txt
本地 Gradio 应用程序
python gradio_app.py
手动推理
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。
- 更新 setuptools
pip install --upgrade setuptools
- 通过以下方式安装其他依赖项
pip install -r requirements.txt