个人主页:IT贫道-CSDN博客
私聊博主:私聊博主加WX好友,获取更多资料哦~
博主个人B栈地址:豹哥教你学编程的个人空间-豹哥教你学编程个人主页-哔哩哔哩视频
目录
1. 下载项目
2. 安装依赖
3. 准备训练好的模型
4. 准备数据目录
5. 执行项目,转换图片
项目地址:GitHub - taki0112/UGATIT: Official Tensorflow implementation of U-GAT-IT: Unsupervised Generative Attentional Networks with Adaptive Layer-Instance Normalization for Image-to-Image Translation (ICLR 2020)
下载后将项目解压到J盘下没有中文路径下:
conda activate python36_cartoon
pip install -i Simple Index tensorflow==1.14
将下载好的模型“100_epoch_selfie2anime_checkpoint.zip”解压得到“100_epoch_selfie2anime_checkpoint”目录,将目录中的“checkpoint”目录复制到“J:\UGATIT-master”目录下:
在“J:\UGATIT-master”目录下创建目录“dataset”目录,在“dataset”目录下创建“selfie2anime”目录,在“selfie2anime”目录下创建“testA”,“testB”目录,“testA”目录存放需要转换成动漫的人脸,“testB”下存放需要转换成人脸的动漫脸。
进入到目录”J:\UGATIT-master”,此目录下打开cmd,执行如下命令:
conda activate python36_cartoon
python main.py --dataset selfie2anime --light False --phase test
注意:selfie2anime 名称需要保持一致,--phase执行是测试数据,--light默认为False使用训练好的模型