JoJoGAN是2021年的一篇关于风格迁移的CVPR,最近看起来了之前的论文,无意之中看到了这一篇,觉得介绍的蛮有意思的,就想着复现一下,实现并不难,理论部分就不在说了,大家看了论文应该都能了解到了,而且百度很多讲解内容,我主要说一下如何实现的。
当然也可以通过colab实现,就是我运行过程中网络不太稳定,所以我就选择了另外一种方式。
1、首先,配置环境,可以根据官网提供的配置文件来安装。
https://github.com/mchong6/JoJoGAN
conda create -n torch python=3.9 -y
conda activate torch
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch -y
2、下载所需要的文件
<