AI绘画神器:DragGAN在线和本地运行的方法

AI工具软件教程11个月前更新 1AiBox
1.6K 0

DragGAN有许多令人兴奋的应用场景,比如你可以使用DragGAN来使动物的嘴张开或闭合,调整汽车的角度,甚至将风景中的山体抬高一些。

DragGAN是一个非常有趣和实用的开源AI图像项目,它使用生成对抗网络(GAN)技术,使用户能够通过简单的拖动操作来改变图像中的静态对象的姿势、形状、表情和方向等。

AI绘画神器:DragGAN在线和本地运行的方法

DragGAN功能演示

 

在线运行和使用DragGAN

如果你只想上手试试看DragGAN的效果,那么在线运行DragGAN是一个更好的选择,我们这里将介绍使用OpenXLab、Hugging Face、以及Google Colab等3个方法来在线使用DragGAN。

 

通过OpenXLab浦源运行

OpenXLab浦源是由「上海人工智能实验室」推出的AI模型社区和托管平台,类似于下文将介绍的Hugging Face。DragGAN的团队在OpenXLab中提供了官方对DragGAN项目的实现,用户可以访问该网站,运行DragGAN测试拖拽效果。

运行地址:https://openxlab.org.cn/apps/detail/XingangPan/DragGAN

AI绘画神器:DragGAN在线和本地运行的方法

通过Hugging Face运行

Hugging Face是一个知名的开源AI模型社区,该平台汇集了各种类型的AI模型、数据集和运行空间。DragGAN项目开源上线后,其官方团队第一时间在Hugging Face上创建了一个项目空间,感兴趣的用户可以随时访问,测试运行效果。

运行地址:https://huggingface.co/spaces/DragGan/DragGan

AI绘画神器:DragGAN在线和本地运行的方法

通过Google Colab使用

Google Colab是由谷歌推出的,允许用户使用Jupyter notebook环境编写、运行和共享Python代码的云端平台。Google Colab是一项免费服务,为用户提供虚拟机,包括访问高性能CPU、GPU和TPU资源,以及预装库,如TensorFlow、PyTorch等。如果你是一位技术人员,那么你可以选择该方法,打开下方的运行地址,然后在设置中选择GPU并运行安装代码和演示代码块即可。

运行地址:https://colab.research.google.com/drive/1mey-IXPwQC_qSthI5hO-LTX7QL4ivtPh?usp=sharing

AI绘画神器:DragGAN在线和本地运行的方法

 

本地安装和使用DragGAN

如果你想要在自己的电脑上本地安装和部署DragGAN,只需根据DragGAN GitHub项目上的指示来操作即可。操作前请先下载安装GitPythonConda,具体步骤如下:

第1步:克隆DragGAN官方的项目

打开命令提示符或终端,使用git命令,git clone https://github.com/XingangPan/DragGAN.git克隆官方项目到你想要的文件夹。

第2步:设置CUDA和安装Python环境

  1. 克隆完毕后,使用cd命令cd DragGAN进入该文件夹
  2. 然后设置conda环境:conda env create -f environment.yml,(注:如果你是运行的Mac,那么请编辑environment.yml文件并删除- cudatoolkit=11.1,因为Mac没有采用英伟达的GPU。另外,Mac还需设置使用CPU运行,在终端中配置export PYTORCH_ENABLE_MPS_FALLBACK=1
  3. 激活conda环境:conda activate stylegan3
  4. 安装相关依赖项:pip install -r requirements

第3步:下载预训练的模型

使用命令python scripts/download_model.py下载

第4步:设置DragGAN GUI界面并运行

使用命令python visualizer_drag_gradio.py下载基于Gradio的界面,下载安装完后,复制终端里面的Running on local URL里面的本地地址。http://127.0.0.1:7860便可以在本地浏览器中运行了。

 

相关内容:

 

相关文章