Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
lin-tianyu authored Oct 26, 2024
1 parent ba51466 commit 7d78d75
Showing 1 changed file with 3 additions and 6 deletions.
9 changes: 3 additions & 6 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -130,8 +130,6 @@ For pre-trained wights of the denoising UNet, run
bash scripts/download_models_lsun_churches.sh
```

### Trained SDSeg Models
> The model weights trained on medical image datasets will be available soon.

## 📄 Scripts
### Training Scripts
Expand Down Expand Up @@ -215,12 +213,11 @@ isbn="978-3-031-72111-3"

## 🔜 TODO List

- [ ] Organizing the inference code. (Toooo redundant right now.)
- [ ] Reimplement SDSeg in OOP. (Elegance is the key!)
- [ ] Add README for multi-class segmentation.
- [ ] Release model weights.
- [ ] Reimplement using diffusers.
- [ ] Reimplement SDSeg in OOP. (Elegance is the key!)
- [ ] Organizing the inference code. (Toooo redundant right now.)
- [ ] Reduce model checkpoint size (no need to save autoencoder's weights).
- [ ] Reimplement using diffusers.

<!-- ## ✨ Star History -->

Expand Down

0 comments on commit 7d78d75

Please sign in to comment.