超强开源一键照片/视频动漫化风格迁移项目—AnimeGANv2使用

今天刚来工作就看到了一个推荐的项目，主要是用于将原始拍摄的照片很快捷地转化为动漫风格的图像，在之前的项目实践中，对于这类型的项目我的操作实践还是比较少的，正好今天有时间就想着拿来把玩一下，看看效果究竟如何。

首先简单看一下成片效果：

 整体上来看还是不错的，我是觉得挺有意思的，所以就想着一探究竟了。

官方项目在这里，首页截图如下所示：

 可以看到：截止目前，已经有超过4.1K的star量了，着实是很出色的了。

 使用所需的基础环境如下：

python 3.6
tensorflow-gpu 1.15.0 (GPU 2080Ti, cuda 10.0.130, cudnn 7.6.0)
opencv
tqdm
numpy
glob
argparse
onnxruntime (If onnx file needs to be run.)

作者也给出来了使用方法，如下：

 感兴趣的话可以自行详细阅读学习，这里就不再赘述了。

官方也给出来宣传海报：

 简单的介绍就到这里了，接下来按照使用指南实际操作一下。

首先下载项目到本地，转化操作命令如下：

python test.py --checkpoint_dir checkpoint/generator_Hayao_weight --test_dir dataset/test/self --save_dir self/self_photo

我在官方项目的dataset/test/目录下面新建了self目录，如下所示：

 在self目录中放入自己想要进行转化测试的图像即可，然后终端执行上面的命令，资源占用还是很高的，启动后电脑直接卡顿了，如下：

 终端输出如下：

 产生的漫画风格的结果图像会自动在results目录下递归创建目录进行存储的，简单看一些样例图片，如下所示：

 感觉对于有特定色差、层次感较强的图像转化效果还是比较明显的，对于最后一张图片转化出来的效果就没有前面的那么明显了。

在我之前的博客中有写的基于焊缝质量检测数据集开发的目标检测项目，这里就突发奇想不知道对于这类型的数据集，模型能否转化成功了，果断拿来尝试一下，操作方法同上，也是在dataset/test/目录下创建新的子目录hanjie，如下所示：

 之后终端执行下面的命令：

python test.py --checkpoint_dir checkpoint/generator_Hayao_weight --test_dir dataset/test/hanjie --save_dir self/hanjie_photo

执行完成后简单看下样例图像：

 可以看到：这里的漫画感还是很明显的。

官方也提供了视频动漫化的转化命令，同样是一键转化，如下：

python video2anime.py --video video/input/demo.mp4 --checkpoint_dir checkpoint/generator_Hayao_weight --output video/output

由于上传的限制这里就不再上传了。

简单截图看下效果： 

感兴趣的话可以自己动手试试还是挺有意思的。