雷达智富

首页 > 内容 > 程序笔记 > 正文

程序笔记

3D内容生成框架threestudio

2024-07-27 19

今天要给大家推荐一个 GitHub 开源项目 threestudio-project/threestudio,该项目在 GitHub 有超过 4.5k Star,用一句话介绍该项目就是:“A unified framework for 3D content generation.”。

ThreeStudio 是一个从文本提示、单个图像和少量图像生成 3D 内容的统一框架。通过提升 2D 文本到图像生成模型,为 3D 内容生成带来新的可能性。

该项目的优点有很多,以下是其中一些介绍:

同一框架下支持多种生成模式,从文本到 3D,图片到 3D,几乎无所不能; 强大的扩展性,以扩展系统为基础,您可以添加自己的扩展,丰富框架的多样性和功能性; 持续激活的更新,作者经常会对新的推理和开源项目进行适配,让框架始终保持最前沿的状态。

以下是一些生成效果:

如何使用

以下是一个在 Ubuntu20.04 系统中测试过可行的安装步骤(其他系统也可根据该步骤试一试):

1、确保你拥有至少 6GB 显存的 NVIDIA 显卡并已安装 CUDA。

2、安装 Python>=3.8。

3、(可选)创建虚拟环境并激活。

4、安装 PyTorch>=1.12。

5、安装 ninja 来加快编译 CUDA 扩展的速度。

6、使用pip install -r requirements.txt来安装依赖项。

具体的使用示例可参考项目 README,比如如果要按一个图片生成 3D 效果,大致代码如下:

python launch.py --config configs/stable-zero123.yaml --train --gpu 0 data.image_path=./load/images/2d_image.png

ThreeStudio GitHub地址:

https://github.com/threestudio-project/threestudio

更新于:1个月前
赞一波!

文章评论

全部评论