Skip to content

Style GAN model compatible with Pytorch Hub

Notifications You must be signed in to change notification settings

ndahlquist/pytorch-hub-stylegan

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

16 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Pytorch Hub StyleGAN

This is a fork of lernapparat/lernapparat to make it available via Pytorch Hub.

(lernapparat/lernapparat itself is a Pytorch implementation of NVlabs/stylegan, using the same / official weights)

Pytorch Hub makes it extremely easy to load the Style GAN model and pretrained weights:

import torch

model = torch.hub.load('ndahlquist/pytorch-hub-stylegan:0.0.1', 'style_gan', pretrained=True)

# done! so easy!

Open this example in Colab: Open Demo in Colab

License

The weights are taken from the reference implementation distributed by NVidia Corporation as Licensed under the CC-BY-NC 4.0 license. As such, the same applies here.

About

Style GAN model compatible with Pytorch Hub

Resources

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Jupyter Notebook 99.4%
  • Python 0.6%