太牛了！这个神奇工具让你的照片秒变宫崎骏漫画

来源丨开源前哨（ID：osfront）

作者丨小秋

简介

AnimeGANv2 是开源项目 AnimeGAN 的改进版本，可以将图片和视频转换成漫画风格的工具。采用的是神经风格迁移 + 生成对抗网络（GAN）的组合，转换速度快。

AnimeGANv2 改进的内容主要包括以下几点：

1. 解决生成图像中高频伪影的问题
2. 易于训练，能达到对应论文中的效果
3. 进一步减少生成器网络的参数数量（生成器大小：8.17Mb），精简版的生成器会更小
4. 使用新的高质量风格数据，尽可能来自 BD 电影

项目地址是：

https://github.com/TachibanaYoshino/AnimeGANv2

用法

依赖

• python 3.6
• tensorflow-gpu 1.15.0
• opencv
• tqdm
• numpy
• glob
• argparse
• onnxruntime

用法

1. 推理

python test.py --checkpoint_dir checkpoint/generator_Hayao_weight --test_dir dataset/test/HR_photo --save_dir Hayao/HR_photo

1. 将视频转换为动画

python video2anime.py --video video/input/お花見.mp4 --checkpoint_dir checkpoint/generator_Hayao_weight --output video/output

1. 训练

• 下载 vgg19

https://github.com/TachibanaYoshino/AnimeGAN/releases/tag/vgg16%2F19.npy

• 下载数据集

https://github.com/TachibanaYoshino/AnimeGAN/releases/tag/dataset-1

• 边缘平滑

python edge_smooth.py --dataset Hayao --img_size 256

• 训练

python train.py --dataset Hayao --epoch 101 --init_epoch 10

• 提取生成器的权重

python get_generator_ckpt.py --checkpoint_dir ../checkpoint/AnimeGANv2_Shinkai_lsgan_300_300_1_2_10_1 --style_name Shinkai

结果

今敏动画风格

宫崎骏动漫风格

新海诚动漫风格

点分享

点点赞

点在看