Skip to content

Latest commit

 

History

History
 
 

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 
 
 

Pytorch-ZOO GNet

Input: no input (random input generated inside of the python script)

Output

Celeb model

Output

Celeb model is converted from pre-trained model of Pytorch-ZOO GAN.

AnimeFace model

Output

The generative networks presented and used here have been trained to generate anime faces and celebrity faces by the company ax Inc. (in 2020).

The training was performed using the framework Pytorch-ZOO GAN. Software license agreement of Pytorch-Zoo GAN: see LICENSE file

Usage

Automatically downloads the onnx and prototxt files on the first run. It is necessary to be connected to the Internet while downloading.

To generate output.png,

$ python3 pytorch-gan.py

You can use --savepath option to change the name of the output file to save.

$ python3 pytorch-gan.py --savepath SAVE_IMAGE_PATH

You can also generate using anime model.

$ python3 pytorch-gan.py -m anime

Reference

Code repo for the Pytorch GAN Zoo project (used to train this model)

Framework

Pytorch

Model Format

ONNX opset=10

Netron

pytorch-gnet-animeface.onnx.prototxt

pytorch-gnet-celeba.onnx.prototxt