AnimeGANv2 正在参加 2021 年度 OSC 中国开源项目评选,请投票支持!
AnimeGANv2 在 2021 年度 OSC 中国开源项目评选 中已获得 {{ projectVoteCount }} 票,请投票支持!
2021 年度 OSC 中国开源项目评选 正在火热进行中,快来投票支持你喜欢的开源项目!
2021 年度 OSC 中国开源项目评选 >>> 中场回顾
AnimeGANv2 获得 2021 年度 OSC 中国开源项目评选「最佳人气项目」 !
授权协议 未知
开发语言 Python 查看源码 »
操作系统 跨平台
软件类型 开源软件
开源组织
地区 不详
投 递 者 Alias_Travis
适用人群 未知
收录时间 2021-11-29

软件简介

AnimeGANv2 是 AnimeGAN 的改进版本,是一个可以讲风景照片/视频转换为动漫图片的工具。

特性:

AnimeGANv2 的改进方向主要包括以下 4 点

  • 解决生成图像中高频伪影的问题。
  • 易于训练,直接达到论文中的效果。
  • 进一步减少生成器网络的参数数量。
  • 使用新的高质量风格数据,尽可能来自 BD 电影。

要求

  • Python 3.6
  • tensorflow-gpu 1.15.0(GPU 2080Ti,cuda 10.0.130,cudnn 7.6.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

2. 将视频转换为动画

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

3. 训练

1. 下载 vgg19

vgg19.npy

2. 下载 Train/Val Photo 数据集

Link

 

3. edge_smooth

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

4. 训练

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

5. 提取生成器的权重

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

展开阅读全文

代码

的 Gitee 指数为
超过 的项目

评论 (0)

加载中
更多评论
暂无内容
暂无内容
暂无内容
0 评论
0 收藏
分享
OSCHINA
登录后可查看更多优质内容
返回顶部
顶部