计算机系统应用教程网站

网站首页 > 技术文章 正文

DragGAN本地安装与使用详细教程,拖拽式修图AI智能补全图片内容

btikc 2025-01-12 11:44:20 技术文章 18 ℃ 0 评论

新一代修图神器DragGAN震撼发布,它是根据指令修图,还能根据内容变换智能补全内容,直接看效果吧,如下图所示。

DragGAN github项目地址:https://github.com/XingangPan/DragGAN,下面讲一下具体安装操作步骤.

1、安装Python

打开Python官网下载链接:https://www.python.org/downloads/,你是什么系统,就下载那个对应版本就可以。

我们下载Python 3.10.11稳定版

下载到电脑上之后,双击运行安装Python,先点击勾选add python.exe to path,在点击install now安装。

2、安装git

git官网地址:https://git-scm.com/download

64位Windows版git程序文件链接:https://github.com/git-for-windows/git/releases/download/v2.41.0.windows.1/Git-2.41.0-64-bit.exe

全部默认安装就可以,一直下一步,完成安装。

3、安装cuda

安装NVIDIA CUDA,打开链接:https://developer.nvidia.com/cuda-11-8-0-download-archive,我们安装11.8版本,如下图选择,

安装程序下载下来之后,直接双击运行安装即可,安装选项选择精简版,安装所需时间较长,安装完成关闭窗口即可。

4、安装Visual Studio

打开微软VS下载链接:https://visualstudio.microsoft.com/zh-hans/vs/community/,点击下载按钮,将安装程序下载到本地电脑后双击安装,按如下界面勾选Python开发、使用C++的桌面开发和右侧Windows 10SDK,注意你的硬盘容量,点击右下角安装按钮安装vs。

5、下载安装DragGAN程序

以安装在D盘为例,在D盘地址栏里输入cmd,回车,启动终端,在终端里输入如下命令,回车

git clone https://github.com/XingangPan/DragGAN.git

然后打开DragGAN文件夹,在地址栏里输入cmd,回车启动终端,在终端里输入如下命令,回车

pip install -r requirements.txt

打开Visual Studio的安装位置,复制如下路径:E:\Microsoft\vs2022\VC\Tools\MSVC\14.36.32532\bin\Hostx64\x64,我的VS是安装在E盘Microsoft\vs2022目录下的,如果你没自定义目录的话,那你的就是在C:\Program Files (x86)目录下,找不到具体路径的话可以搜一下MSVC这个文件夹,依次打开上述路径,复制路径地址。然后用VS CODE或是其它编辑工具打开DragGAN文件夹内torch_utils\custom_ops.py这个文件,找到第31行按下图修改成你的实际路径地址,保存关闭custom_ops.py这个文件。这里不修改的话运行软件的时候会提示MSVC/GCC/CLANG错误。

6、下载模型

下载预训练的 StyleGAN2模型,接着在上面的终端里面输入如下命令,回车

python scripts/download_model.py

程序会自动下载8个模型到checkpoints文件夹内,如果没有checkpoints这个文件夹的话你可以手动创建一个。如果你网速不好的话可以到我网盘里下载,还有2个stylegan_human和1个lhq模型我也一并上传了,

7、程序运行体验

接着在上面的终端里输入如下命令,回车,第一次打开的时候需要加载模型会很慢。

scripts\gui.bat

操作界面功能比较简单,如上图所示,点击add point在图片上点击想要变化的区域,然后点击start就开始调整图片了。点击save image可以保存修改后的图片。这个软件对电脑显卡要求还是很高的,使用1024的模型一点开始就卡住不动了,只能用小尺寸的模型,小尺寸的模型显存也经常爆。有对软件感兴趣的可以去试试。

原文:https://dyss.top/1111

Tags:

本文暂时没有评论,来添加一个吧(●'◡'●)

欢迎 发表评论:

最近发表
标签列表