/AI-Avatar-Creater

Use pytorch and GAN to create your own avatar

Primary LanguageJavaScript

AI-Avatar-Creater

Use GAN (Generative adversarial network) to generate your anime avatar and style transfer!

Note: This codebase was developed in 2018, which is out of date in the 2020s. This code is only for reference and will no longer be updated. Today, a diffusion-based model is recommended for image generation tasks.

Contributors

  • Algorithm & design：@creeper121386
  • Web：@Benjaminyuan
  • Deployment：@ThinCats

Introduction

Real-time style transfer

This is paper

Use GAN to generate anime avatars

Build

To build, you need:

  • NVIDIA with Cuda-toolkit
  • PyTorch
$ pip install PyTorch
  • Flask (for web view)
$ pip install flask
  • Mataplotlib

Run

  1. Go to hackweek directory, and then run using python manage.py
  2. It will download the PyTorch model automatically; wait patiently.
  3. If successful, it should appear the basic server information. Go to http://localhost:5000 to have fun

Future

Docker image (Still waited to be optimized)