merve HF staff commited on
Commit
2f1b55b
1 Parent(s): 8a4369c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +18 -1
README.md CHANGED
@@ -8,8 +8,25 @@ datasets:
8
  ---
9
  ## Model description
10
  Anime face generator model using [TensorFlow DCGAN example](https://www.tensorflow.org/tutorials/generative/dcgan).
 
11
  ## Training and evaluation data
12
- Model is trained on [anime faces dataset](https://huggingface.co/datasets/merve/anime-faces).
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
13
  ## Intended use and biases
14
  This model is not intended for production.
15
 
 
8
  ---
9
  ## Model description
10
  Anime face generator model using [TensorFlow DCGAN example](https://www.tensorflow.org/tutorials/generative/dcgan).
11
+
12
  ## Training and evaluation data
13
+ Model is trained on [anime faces dataset](https://huggingface.co/datasets/merve/anime-faces). The dataset consists of 21551 anime faces scraped from www.getchu.com, which are then cropped using the anime face detection algorithm [here](https://github.com/nagadomi/lbpcascade_animeface). All images are resized to 64 * 64 for the sake of convenience. The model takes a noise as input and then Conv2DTranspose is used to do upsampling. If you want to pass this to another discriminator, the output shape consists of 28x28 images.
14
+
15
+ ## How to use this model
16
+ You can use this model to generate new anime faces. If you want to continuously train, use with [discriminator](https://huggingface.co/merve/anime-faces-discriminator) using `tf.GradientTape()` as mentioned in the DCGAN tutorial.
17
+
18
+ ```
19
+ from huggingface_hub import from_pretrained_keras
20
+
21
+ model = from_pretrained_keras("merve/anime-faces-generator")
22
+ ```
23
+
24
+ You can generate examples using a noise.
25
+
26
+ ```
27
+ seed = tf.random.normal([number_of_examples_to_generate, noise])
28
+ predictions = model(seed, training=False) # inference mode
29
+ ```
30
  ## Intended use and biases
31
  This model is not intended for production.
32