Update README.md
Browse files
README.md
CHANGED
|
@@ -21,3 +21,11 @@ This model repo contains checkpoints trained on Cityscapes and ADE20K datasets u
|
|
| 21 |
For usage instructions, please refer to our <a href="https://github.com/boschresearch/ALDM">Github</a>.
|
| 22 |
</div align="center">
|
| 23 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 21 |
For usage instructions, please refer to our <a href="https://github.com/boschresearch/ALDM">Github</a>.
|
| 22 |
</div align="center">
|
| 23 |
|
| 24 |
+
## Model information
|
| 25 |
+
[ade20k_step9.ckpt](ade20k_step9.ckpt) and [cityscapes_step9.ckpt](cityscapes_step9.ckpt) are pretrained diffusion model weights for inference.
|
| 26 |
+
|
| 27 |
+
[encoder_epoch_50.pth](encoder_epoch_50.pth), [decoder_epoch_50_20cls.pth](decoder_epoch_50_20cls.pth) and [decoder_epoch_50_151cls.pth](decoder_epoch_50_151cls.pth)
|
| 28 |
+
are segmentation models used for disciminator intialization in training,
|
| 29 |
+
which are adopted from pretrained weights from [here](https://github.com/CSAILVision/semantic-segmentation-pytorch).
|
| 30 |
+
|
| 31 |
+
|