BrushNet即插即用的局部图片修复项目

BrushNet 是一种基于扩散的文本引导图像修复模型,可以即插即用到任何预先训练的扩散模型中。具体是做什么的,你可以先看下面的视频

项目的github主页:https://github.com/TencentARC/BrushNet

BrushNet即插即用的局部图片修复项目

本地部署教程

克隆仓库文件到本地

可以直接通过git命令来克隆

git clone https://github.com/TencentARC/BrushNet.git

BrushNet即插即用的局部图片修复项目

也可以通过下载zip压缩包到本地解压

BrushNet即插即用的局部图片修复项目

不管通过那种方式,你最后都会得到像下面的文件夹

BrushNet即插即用的局部图片修复项目

新建虚拟环境+安装Python库

进入到该文件夹中,在路径栏输入“cmd”命令,回车,会打开一个cmd命令窗口。

BrushNet即插即用的局部图片修复项目

在cmd窗口中先后执行如下的命令:

用conda创建一个名为“brushnet”的虚拟环境

conda create -n brushnet python=3.9 -y

BrushNet即插即用的局部图片修复项目

激活该虚拟环境

conda activate brushnet

升级pip

python -m pip install --upgrade pip

安装torch三件套

pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu116

BrushNet即插即用的局部图片修复项目

安装扩散器和依赖包

您可以使用以下命令安装扩散器:

pip install -e .

BrushNet即插即用的局部图片修复项目

安装项目所需的依赖包

接着继续执行如下的命令,安装项目所需的依赖包

cd examples/brushnet/
pip install -r requirements.txt

BrushNet即插即用的局部图片修复项目

原创文章,作者:朋远方,如若转载,请注明出处:https://caovan.com/brushnetjichajiyongdejubutupianxiufuxiangmu/.html

(2)
打赏 微信扫一扫 微信扫一扫
朋远方的头像朋远方
上一篇 2024年3月28日 下午6:20
下一篇 2024年4月6日 上午8:56

相关推荐

发表回复

登录后才能评论