Muhammad Naufal Rizqullah commited on
Commit
439c972
·
1 Parent(s): eb42124

change readme and change weight model

Browse files
README.md CHANGED
@@ -1,49 +1,13 @@
1
- # Conditional GAN Shoe, Sandal, Boot
2
- =====================================
3
-
4
- ## Project Overview
5
- ---------------
6
-
7
- This project implements a Conditional Generative Adversarial Network (CGAN) using the Conditional WGAN-GP architecture to generate images of shoes, sandals, and boots.
8
-
9
- ### Key Features
10
- ------------
11
-
12
- * **Conditional WGAN-GP architecture**: for generating images of shoes, sandals, and boots
13
- * **Trained on a limited dataset**: due to VRAM constraints
14
- * **Model architecture and hyperparameters optimized**: for Kaggle environment with 15GB VRAM
15
- * **Embedding used instead of one-hot encoding**: for training labels to avoid using 0 on labels
16
- * **Implemented using PyTorch Lightning framework**
17
-
18
- ## Training Details
19
- ---------------
20
-
21
- ### Training Epochs
22
- ----------------
23
-
24
- * Approximately 1000 epochs
25
-
26
- ### Model Architecture Compromises
27
- -----------------------------
28
-
29
- * **Reduced Size of latent space (z dim) and Embedding**: due to VRAM limitations
30
- * **Limited features for generator and critic networks**: due to VRAM limitations
31
- * **Image size limitations**: due to VRAM limitations
32
- * **Dataset used**: [Shoe vs Sandal vs Boot Image Dataset (15K Images)](https://www.kaggle.com/datasets/hasibalmuzdadid/shoe-vs-sandal-vs-boot-dataset-15k-images)
33
-
34
- ## Results
35
- ---------
36
-
37
- Despite the architectural compromises, the model produces reasonable results. However, the quality of the generated images may not be optimal due to the limited dataset and VRAM constraints.
38
-
39
- ## Demo
40
- -------------
41
- The demo of this project is deployed on Hugging Face's model hub and uses the Gradio framework to provide a user-friendly interface for interacting with the model. You can try out the demo by visiting [this link](https://huggingface.co/spaces/SkylarWhite/57894).
42
-
43
-
44
- ## Future Work
45
- -------------
46
-
47
- * **Experiment with larger datasets and more complex model architectures**
48
- * **Investigate alternative optimization techniques to improve model performance**
49
- * **Explore other applications of Conditional GANs in computer vision**
 
1
+ ---
2
+ title: '57894'
3
+ emoji: 🐠
4
+ colorFrom: indigo
5
+ colorTo: purple
6
+ sdk: gradio
7
+ sdk_version: 4.41.0
8
+ app_file: app.py
9
+ pinned: false
10
+ license: mit
11
+ ---
12
+
13
+ Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
weights/{epoch=999-step=96000.ckpt → epoch=957-step=1164300.ckpt} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:93409ee79fbe7ecfbcd95fe775a7625408e088624c0e80153e14c234c93d8132
3
- size 116330608
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1602e9cb80fcfbc39ed2c5c74873744a546cbab346730aaa0f25b02140bc1d2f
3
+ size 157799852