File size: 1,292 Bytes
4d5ccc7 76c9db5 4d5ccc7 d203629 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
---
license: apache-2.0
library_name: pytorch
tags:
- biggan
datasets:
- ImageNet
---
## Model description
This is an op-for-op PyTorch reimplementation of DeepMind's BigGAN model with the pre-trained weights from DeepMind [biggan-deep-128](https://tfhub.dev/deepmind/biggan-deep-128/1).
## Training and evaluation data
Model is trained on [ImageNet dataset](https://tfhub.dev/s?dataset=imagenet-ilsvrc-2012-cls). The dataset consists of 10000 classes. All images are resized to 64 * 64 for the sake of convenience. The model takes noise as input and then Conv2DTranspose is used to do upsampling. The output shape consists of 128, 256, or 512 images depending on the model.
## How to use this model
You can use this model to generate new images.
```
import torch
from pytorch_pretrained_biggan import (BigGAN, one_hot_from_names, truncated_noise_sample,
save_as_images, display_in_terminal)
model = BigGAN.from_pretrained('biggan-deep-256')
```
You can generate examples using a noise vector.
```
with torch.no_grad():
output = model(noise_vector, class_vector, truncation)
```
## Intended use and biases
This model is not intended for production.
### Generated images
### Credits
@thomwolf
Thomas Wolf
@vfdev-5
vfdev |