图像动漫风格转换：AnimeGANv2 + Flask

图像动漫风格转换：AnimeGANv2 + Flask

项目介绍

本项目是基于 animegan2-pytorch 的 AnimeGANv2 配合 Flask 在网页端展示动漫化的简单 demo。

环境配置

• Python 3.x

• PyTorch + Torchvision

  可以不装 cuda。

• Pillow

  可以通过 pip install pillow 安装。

• opencv

  可以通过 pip install opencv-python 安装。

• flask

  可以通过 pip install flask 安装。

文件结构

  ├── images: 存放测试图片的文件夹
  ├── static：存放静态文件的文件夹，包括生成的结果图片 `img_origin.jpg` 和 `img_output.jpg`
  ├── templates：存放 html 模板的文件夹
  ├── weights：存放模型权重的文件夹，可以根据文件夹下的权重文件更换不同的动漫效果
  ├── app.py：运行的主函数
  ├── model.py：搭建网络模型的文件
  └── requirements.txt：环境配置文件

运行方式

直接运行 app.py。

使用说明

• 一次只能动漫化一张图片。
• 为了在页面中展示完整，代码会将输入图片 resize 到固定大小，并和生成的图片进行横向拼接，从而在网页端进行对比展示。
• 如果图片过大（> 2 MB）则无法接收图片。可以在 templates/base.html 的第 46 行将数字 2（2 MB）换成一个更大的数字。

项目地址

相关代码通过 https://gitee.com/tongsen99/anime-gan-flask 访问。

---

AnimeGAN 相关背景

AnimeGANv2 结果展示