如何安装和使用DragGAN？在线和本地运行的4种方法

AI教程1年前 (2023)更新 AI小集

0 9

DragGAN是一个热门的开源AI图像项目和研究，可以通过拖动对图像中的静态对象的姿势、形状、表情和方向等进行灵活而精确的控制，可以实现比如将动物的嘴张开、将汽车的角度更改、将风景中的山体拔高等操作。近期，DragGAN的团队在GitHub上线并开源了其官方的实现项目，本篇文章将介绍如何运行、安装和使用DragGAN，包含在线运行和本地使用的多种方法。

文章目录：

在线运行和使用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

通过Hugging Face运行

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

运行地址：https://huggingface.co/spaces/DragGan/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

本地安装和使用DragGAN

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

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

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

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

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