【可视化学习】70-图片转3D模型
发表于:2024-06-17 |

前言

今天给大家带来一个库–TripoSR,我们可以用来简单来使用图片生成模型,虽然效果和那种商业的没法比,但我们平时使用效果也算是可以的了。话不多说,我们这就开始。

温馨提示

你跟着操作可能需要点科技,这里大家可以找找我之前的文章

安装

拉取代码

1
git clone https://github.com/VAST-AI-Research/TripoSR.git

python环境

因为代码都是python写的,所以你需要安装一个python环境

配置

1
2
3
4
5
6
#设置python环境
conda create -n tsr python=3.10
conda activate tsr

#安装pytorch
pip3 install torch torchvision torchaudio

安装相关依赖

1
2
3
4
pip install --upgrade setuptools

cd TripoSR
pip install -r requirements.txt

安装gradio

1
pip install gradio

模型地址

TripoSR模型

TripoSR模型地址:https://huggingface.co/stabilityai/TripoSR
模型地址

下载config.yaml和model.ckpt,两个文件,点击VERIFIED值转到以下页面,并复制该值。
下载地址
当前版本

VERIFIED值 :c1cf7716aed5aa6c1c5e174657791ef0e1327bde,注意不同版本时该值可能不同。本案所讲解的只代表这一阶段下载时的VERIFIED值。将这两个文件放置到下面的文件夹中。设置位置如下:

/root/.cache/huggingface/hub/models–stabilityai–TripoSR

在这下面创建以下几个文件:
创建文件

blobs文件夹,保持为空

refs文件夹,下面一个main文件,main文件中放置下载时的VERIFIED值,当前下载时的VERIFIED值为:c1cf7716aed5aa6c1c5e174657791ef0e1327bde

snapshosts文件夹,在这下面以VERIFIED值为文件夹创建一个文件夹,名字为:c1cf7716aed5aa6c1c5e174657791ef0e1327bde,然后把模型放在这个文件夹下面

dino-vitb16

该模型位于:https://github.com/facebookresearch/dino

目录结构如下:/root/.cache/huggingface/hub/models–facebook–dino-vitb16,目录结构同(1)

拿到dino-vitb16当前版本的VERIFIED值,然后与(1)的方法一样,创建相应的文件夹,把模型和文件放入相应的文件夹。

u2net.onnx

一般情况下,运行gradio_app.py会自动下载该模型,如无法正常下载,你可以手动将该模型copy到:
/root/.u2net/u2net.onnx

运行文件

修改运行文件:gradio_app.py

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
    interface.launch(
auth=(args.username, args.password) if (args.username and args.password) else None,
share=args.share,
server_name="0.0.0.0" if args.listen else None,
server_port=args.port
)

#修改为:

interface.launch(
auth=(args.username, args.password) if (args.username and args.password) else None,
share=True,
server_name="192.168.29.221",#请填写实际的服务器IP地址。
server_port=args.port
)

图形页面请运行下面的指令:

1
python gradio_app.py

效果
效果

常见问题

1、No module named ‘trimesh’
pip install trimesh==2.38.39

2、No module named ‘omegaconf’
pip install omegaconf

3、No module named ‘skimage’
pip install scikit-image

4、No module named ‘shapely’
pip install shapely

5、AttributeError: module ‘numpy’ has no attribute ‘bool’.目前适用版本numpy==1.23.5
pip uninstall numpy
pip install numpy==1.23.5

结语

本篇文章就到这里了,更多内容敬请期待,债见~

上一篇:
window.atob的使用问题
下一篇:
【可视化学习】69-从入门到放弃WebGL(七)