Commit
•
8acc560
1
Parent(s):
8820108
Update README.md
Browse files
README.md
CHANGED
@@ -16,30 +16,26 @@ inference: true
|
|
16 |
should probably proofread and complete it, then remove this comment. -->
|
17 |
|
18 |
|
19 |
-
# LoRA text2image fine-tuning
|
20 |
|
21 |
|
22 |
|
23 |
These are LoRA adaption weights for stabilityai/stable-diffusion-xl-base-1.0.
|
24 |
Original dataset: https://www.kaggle.com/datasets/tourist55/alzheimers-dataset-4-class-of-images.
|
25 |
|
26 |
-
With this model you can generate MRI of the brain of 4 classes:
|
27 |
1) NonDemented
|
28 |
2) VeryMildDemented
|
29 |
3) MildDemented
|
30 |
4) ModerateDemeneted
|
31 |
|
32 |
-
For generation use following corresponding prompts (or any other modification)
|
33 |
|
34 |
1) MRI of NonDemented brain
|
35 |
2) MRI of VeryMildDemented brain
|
36 |
3) MRI of MildDemented brain
|
37 |
4) MRI of ModerateDemeneted brain
|
38 |
|
39 |
-
|
40 |
-
|
41 |
-
You can find some example images in the following.
|
42 |
-
|
43 |
![img_0](./image_0.png)
|
44 |
![img_1](./image_1.png)
|
45 |
![img_2](./image_2.png)
|
|
|
16 |
should probably proofread and complete it, then remove this comment. -->
|
17 |
|
18 |
|
19 |
+
# BRAIN with Alzheimer's MRI generation (LoRA text2image fine-tuning)
|
20 |
|
21 |
|
22 |
|
23 |
These are LoRA adaption weights for stabilityai/stable-diffusion-xl-base-1.0.
|
24 |
Original dataset: https://www.kaggle.com/datasets/tourist55/alzheimers-dataset-4-class-of-images.
|
25 |
|
26 |
+
With this model, you can generate an MRI of the brain of 4 classes:
|
27 |
1) NonDemented
|
28 |
2) VeryMildDemented
|
29 |
3) MildDemented
|
30 |
4) ModerateDemeneted
|
31 |
|
32 |
+
For generation use the following corresponding prompts (or any other modification)
|
33 |
|
34 |
1) MRI of NonDemented brain
|
35 |
2) MRI of VeryMildDemented brain
|
36 |
3) MRI of MildDemented brain
|
37 |
4) MRI of ModerateDemeneted brain
|
38 |
|
|
|
|
|
|
|
|
|
39 |
![img_0](./image_0.png)
|
40 |
![img_1](./image_1.png)
|
41 |
![img_2](./image_2.png)
|