前言
今天给大家带来一个库–TripoSR,我们可以用来简单来使用图片生成模型,虽然效果和那种商业的没法比,但我们平时使用效果也算是可以的了。话不多说,我们这就开始。
温馨提示
你跟着操作可能需要点科技,这里大家可以找找我之前的文章
安装
拉取代码
1 | git clone https://github.com/VAST-AI-Research/TripoSR.git |
python环境
因为代码都是python写的,所以你需要安装一个python环境
配置
1 | #设置python环境 |
安装相关依赖
1 | pip install --upgrade setuptools |
安装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 | interface.launch( |
图形页面请运行下面的指令:
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
结语
本篇文章就到这里了,更多内容敬请期待,债见~