Jezia commited on
Commit
4d5ccc7
1 Parent(s): fa84773

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +30 -0
README.md ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ library_name: keras
4
+ tags:
5
+ - biggan
6
+ datasets:
7
+ - ImageNet
8
+ ---
9
+ ## Model description
10
+ 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).
11
+ ## Training and evaluation data
12
+ 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.
13
+ ## How to use this model
14
+ You can use this model to generate new images.
15
+ ```
16
+ import torch
17
+ from pytorch_pretrained_biggan import (BigGAN, one_hot_from_names, truncated_noise_sample,
18
+ save_as_images, display_in_terminal)
19
+
20
+ model = BigGAN.from_pretrained('biggan-deep-256')
21
+ ```
22
+ You can generate examples using a noise vector.
23
+ ```
24
+ with torch.no_grad():
25
+ output = model(noise_vector, class_vector, truncation)
26
+ ```
27
+ ## Intended use and biases
28
+ This model is not intended for production.
29
+ ### Generated images
30
+ ![Example](./example.png)