Commit
·
61bf487
1
Parent(s):
b1317ab
Create README.md
Browse files
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 |
+

|