本文简单记录介绍一下FreeDrag。
FreeDrag是一种新的交互式基于点的图像编辑框架,与DragGAN相比摆脱了繁琐和不稳定的点跟踪过程。
一种新的交互式基于点的图像编辑框架,基于StyleGAN2,采用基于特征的方法,能够在具有相似结构、精细细节或多点目标的非常具有挑战性场景下实现稳定的基于点的编辑。
相比DragGAN在点跟踪时有丢失和处理混乱等问题,使用新的方法使得模型在更有挑战性的场景下仍然取得更佳的效果。
下载项目源码:git clone --depth=1 https://github.com/LPengYang/FreeDrag
安装StyleGAN2预训练模型:bash download_models.sh
本地运行:CUDA_LAUNCH_BLOCKING=1 python FreeDrag_gradio.py
项目的路线图中还规划了基于Diffusion的FreeDrag和能够处理3D图形的框架,值得期待。
参考文章:
本文仅作为个人学习记录。
本文与HBlog保持同步。