marcelhuber commited on
Commit
61bf487
·
1 Parent(s): b1317ab

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +42 -0
README.md ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ datasets:
3
+ - marcelhuber/KermanyV3_original
4
+ language:
5
+ - en
6
+ tags:
7
+ - medical
8
+ ---
9
+ # KermanyV3 GAN Model
10
+
11
+ ## Description
12
+ The KermanyV3 GAN is a Generative Adversarial Network trained on the KermanyV3_original dataset to generate synthetic medical images.
13
+
14
+ ## Files Included
15
+ - `fakes001020.png`: Sample of synthetic images generated by the GAN at snapshot 001020.
16
+ - `log.txt`: Log file containing training progress and details.
17
+ - `metric-fid50k_full.jsonl`: JSONL file with FID (Fréchet Inception Distance) metric scores for evaluating image quality.
18
+ - `metric-kid50k_full.jsonl`: JSONL file with KID (Kernel Inception Distance) metric scores for evaluating image quality.
19
+ - `network-snapshot-001020.pkl`: Snapshot of the GAN's trained parameters at iteration 001020.
20
+ - `stats.jsonl`: JSONL file containing statistics and metrics related to the training process.
21
+ - `training_options.json`: JSON file detailing the training hyperparameters and options.
22
+
23
+ ## Dataset Used
24
+ KermanyV3_resized
25
+
26
+ ## Model Performance
27
+ - FID Score: 8.745657893911403
28
+ - KID Score: 0.005549223578578
29
+
30
+ ## Training Details
31
+ - Training Iterations: 001020
32
+ - Training Time: 11h 06m
33
+ - Architecture: StyleGAN2-ADA-PyTorch
34
+ - Batch Size: 16
35
+
36
+ ## Usage Instructions
37
+ 1. Load the pre-trained GAN parameters from `network-snapshot-001020.pkl`.
38
+ 2. Use the generator to generate synthetic images.
39
+ 3. To evaluate image quality, refer to `metric-fid50k_full.jsonl` for FID scores and `metric-kid50k_full.jsonl` for KID scores.
40
+ 4. For additional training information, refer to `log.txt`, `stats.jsonl`, and `training_options.json`.
41
+
42
+ ![Generated Image](fakes001020.png)