在上篇中,树先生教大家如何正确部署 DragGAN 项目,实现自由拖拽式 P 图。
但可惜只能使用项目预置的一些图片,本篇教大家如何利用该项目自由编辑修改任何图片。
这里主要使用到 PTI 项目,可以将你自定义的图片训练成 StyleGAN 潜空间模型,从而实现任何图片的编辑修改。
环境准备
这里我们还是继续选择 AutoDL 云平台,使用 Python 3.8,CUDA 11.8 的镜像,这个镜像的环境满足项目要求。
下载源码
git clone https://github.com/danielroich/PTI.git
# 安装依赖
cd PTI && pip install -r requirements.txt