ControlGlaucoma / README.md
ZihengWang's picture
rename to controlnet_* and clean segman filename
04b995f verified
|
Raw
History Blame Contribute Delete
883 Bytes
---
license: apache-2.0
tags:
- controlnet
- stable-diffusion
- fundus
- glaucoma
- medical-imaging
---
# ControlGlaucoma — Pre-trained Weights
Pre-trained weights for [ControlGlaucoma](https://github.com/WANG-ZIHENG/ControlGlaucoma) — vCDR-controllable ControlNet on Stable Diffusion 2.1 for SLO fundus generation.
| Folder | Description |
|---|---|
| `controlnet_vpred/` | Main weight. ControlNet on SD 2.1 (v-prediction). |
| `controlnet_eps/` | ε-prediction variant (compatible with SD 2.1-base, 512 px). |
| `segman_b/` | SegMAN-B segmentation model used as the reward signal during training. |
```bash
huggingface-cli download ZihengWang/ControlGlaucoma \
--local-dir ./checkpoints \
--local-dir-use-symlinks False
```
See the [GitHub repository](https://github.com/WANG-ZIHENG/ControlGlaucoma) for code and training / inference commands.
## License
Apache-2.0.