Files changed (1) hide show
  1. README.md +32 -0
README.md ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ tags:
4
+ - gan
5
+ - stylegan2
6
+ - discriminator
7
+ - image-generation
8
+ - nvidia
9
+ - ffhq
10
+ ---
11
+
12
+ # StyleGAN2 Discriminator (FFHQ, NVIDIA)
13
+
14
+ This model is the **discriminator** component of **StyleGAN2**, based on the original architecture introduced by NVIDIA and trained on the [FFHQ dataset](https://github.com/NVlabs/ffhq-dataset). The model is designed to distinguish between real and fake images, and is typically used in generative adversarial networks (GANs) for high-resolution face synthesis.
15
+
16
+ ## 🧩 Model Details
17
+
18
+ - **Architecture**: StyleGAN2 Discriminator (NVIDIA)
19
+ - **Framework**: PyTorch
20
+ - **Trained on**: FFHQ (Flickr-Faces-HQ)
21
+ - **Use case**: Discriminator in GAN training or standalone real/fake image classification
22
+
23
+ ## 🛠️ How to Use
24
+
25
+ ```python
26
+ from huggingface_hub import hf_hub_download
27
+ import torch
28
+
29
+ # Download the model
30
+ model_path = hf_hub_download(repo_id="mukhbiir/StyleGAN2_Discriminator", filename="model.pt")
31
+ model = torch.load(model_path)
32
+ model.eval()