如何安裝和使用DragGAN?在線和本地運行的4種方法

AI教程1年前 (2023)發佈新公告 AI管理員
5 0

DragGAN是一個熱門的開源AI圖像項目和研究,可以通過拖動對圖像中的靜態對象的姿勢、形狀、表情和方向等進行靈活而精確的控制,可以實現比如將動物的嘴張開、將汽車的角度更改、將風景中的山體拔高等操作。近期,DragGAN的團隊在GitHub上線並開源了其官方的實現項目,本篇文章將介紹如何運行、安裝和使用DragGAN,包含在線運行和本地使用的多種方法。

如何安裝和使用DragGAN?在線和本地運行的4種方法

文章目錄:

  • 在線運行和使用DragGAN
    • 通過OpenXLab浦源運行
    • 通過Hugging Face運行
    • 通過Google Colab使用
  • 本地安裝和使用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

如何安裝和使用DragGAN?在線和本地運行的4種方法

通過Hugging Face運行

Hugging Face是一個知名的開源AI模型社區,該平台彙集了各種類型的AI模型、數據集和運行空間。DragGAN項目開源上線後,其官方團隊第一時間在Hugging Face上創建了一個項目空間,感興趣的用戶可以隨時訪問,測試運行效果。

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

如何安裝和使用DragGAN?在線和本地運行的4種方法

通過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?在線和本地運行的4種方法

本地安裝和使用DragGAN

如果你想要在自己的電腦上本地安裝和部署DragGAN,只需根據DragGAN GitHub項目上的指示來操作即可。操作前請先下載安裝Git、Python和Conda,具體步驟如下:

第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便可以在本地瀏覽器中運行了。

以上便是AI工具集爲大家整理的如何運行DragGAN的方法,如果你在安裝和使用過程中有何問題,可以在評論區留言或着在DragGAN的官方GitHub項目尋求幫助。

© 版權聲明

相關文章

暫無評論

暫無評論...