Skip to content

Commit fe138f8

Browse files
committed
appreciation
1 parent 691324b commit fe138f8

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

README.md

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,12 @@
55

66
Implementation of <a href="https://arxiv.org/abs/2211.00611">MedSegDiff</a> in Pytorch - SOTA medical segmentation out of Baidu using DDPM and enhanced conditioning on the feature level, with filtering of features in fourier space.
77

8+
## Appreciation
9+
10+
- <a href="https://stability.ai/">StabilityAI</a> for the generous sponsorship, as well as my other sponsors out there
11+
12+
- <a href="https://github.com/isamu-isozaki">Isamu</a> and <a href="https://github.com/dsbuddy">Daniel</a> for adding a training script for a skin lesion dataset!
13+
814
## Install
915

1016
```bash
@@ -50,10 +56,6 @@ accelerate launch driver.py --mask_channels=1 --input_img_channels=3 --image_siz
5056
```
5157
If you want to add in self condition where we condition with the mask we have so far, do --self_condition
5258

53-
## Appreciation
54-
55-
- <a href="https://stability.ai/">StabilityAI</a> for the generous sponsorship, as well as my other sponsors out there
56-
5759
## Todo
5860

5961
- [x] some basic training code, with Trainer taking in custom dataset tailored for medical image formats - thanks to <a href="https://github.com/isamu-isozaki">@isamu-isozaki</a>

0 commit comments

Comments
 (0)