DragGAN 上手教程:5分钟搞定零代码拖拽图像编辑
【免费下载链接】DragGANOfficial Code for DragGAN (SIGGRAPH 2023)项目地址: https://gitcode.com/GitHub_Trending/dr/DragGAN
这篇文章解决一件事:把 DragGAN 装进你的机器,用拖拽控制点编辑图像。跟着做,5分钟看到第一张图。
动手前:3件事
三件事:机器、环境、权重。齐了就能跑。
- 机器:NVIDIA 显卡(带 CUDA),或 Mac M1/M2(纯 CPU,会慢些)。
- 环境:克隆仓库,按仓库里的 environment.yml 搭好:
git clone https://gitcode.com/GitHub_Trending/dr/DragGAN cd DragGAN conda env create -f environment.yml conda activate stylegan3 && pip install -r requirements.txt- 权重:
python scripts/download_model.pyStyleGAN2 权重会落到 checkpoints/ 目录。人像权重是另一个 pkl,需手动放进同一目录。
第一次出图:最短路径
一条命令启动网页版:
python visualizer_drag_gradio.py浏览器自动打开,此时你已经有了第一张图——页面加载完就生成了一张 512×512 的图像。左侧面板有 Pretrained Model 和 Seed,改 Seed 数字,图像立刻换一张。5分钟,看到第一张图。
拖拽手感拆解:控制点怎么摆
红点是"抓手",蓝点是"目的地",各点一次就组成一条拖拽指令。
顺序固定:
- 点 Add Points,进入加点模式。
- 第一次点击是起点(红):放在你要挪的部位上,比如狮子头部。
- 第二次点击是终点(蓝):想让它挪到的位置。
- 点 Start,图像沿 GAN 流形逐渐变形;随时可 Stop 暂停。
手感三件事:
- 红点按在"抓手"上,不是画面正中央,而是你想控制的那块。
- 蓝点拉多远,物体就挪多远。第一次拉不到画面宽度的五分之一,感觉太猛就少走点。
- 一次别超过3对点。每轮收敛后看一眼效果,再决定加不加。
- 背景不想动?用 Edit Flexible Area 画个区域,蒙住的地方保持不动。
- 效果不满意?Reset Points,把一次大改拆成两次小改。
💡 优化过程中红点会跟着物体走,盯住它,跟对了就没跑偏。
拿它做点有意思的事
两个最常见的需求,各给你走一遍。
人像姿态不对,改过来
想达到什么效果:图里主体歪了,或者你想换个视线方向。
具体几步:Pretrained Model 里选人像权重(StyleGAN-Human 权重下载后放进 checkpoints/ 目录)。红点按在人脸,蓝点指到新方向。点 Start 等收敛,不够再加一对点。
得到什么:人像的姿态和视线整体挪位,背景、光照保持一致,不"糊"。
真实照片改不动,先反演
想达到什么效果:不想编辑生成图,想把实拍照片里的物体挪个位置。
具体几步:DragGAN 只吃潜码。先用 PTI 反演法把照片转成潜码——stylegan_human/ 目录里就带现成的 run_pti.py。把对应权重和潜码加载进界面,再照常拖拽。
得到什么:实拍照片变成可拖的,物体挪个位,还是实拍质感。
翻车自救手册
跑它容易碰到的问题,压缩成5条。
- 症状:界面卡顿,每步优化等好几秒。处方:确认显卡真在用;没有 CUDA 会落到 CPU,天然慢,减少控制点数量。
- 症状:启动时 checkpoints/ 报找不到文件。处方:没跑下载脚本,或没下完整。重跑
python scripts/download_model.py,确认目录里有 pkl。 - 症状:换了模型或 Seed,刚摆的点没了。处方:设计如此,换权重、换 Seed 都会重置图像和控制点,重新摆点。
- 症状:拖完整张图糊掉、变形。处方:一次拉太远或点太多。Reset Points,改小步幅分次摆。
- 症状:Windows 下
sh scripts/gui.sh报命令不存在。处方:桌面 GUI 面向 Linux/macOS;Windows 直接跑网页版python visualizer_drag_gradio.py。
从克隆仓库到拖出第一张图,全程就四步:装环境、下权重、起网页、摆点按 Start。现在打开终端,5分钟后你会看到图像跟着你点的方向动。
【免费下载链接】DragGANOfficial Code for DragGAN (SIGGRAPH 2023)项目地址: https://gitcode.com/GitHub_Trending/dr/DragGAN
创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考