Skip to content

Commit 1257c4b

Browse files
committed
Add to README
1 parent f2e3f89 commit 1257c4b

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

README.md

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ Segmentation based on [PyTorch](https://pytorch.org/).**
2121
The main features of the library are:
2222

2323
- Super simple high-level API (just two lines to create a neural network)
24-
- 11 encoder-decoder model architectures (Unet, Unet++, Segformer, ...)
24+
- 12 encoder-decoder model architectures (Unet, Unet++, Segformer, DPT, ...)
2525
- 800+ **pretrained** convolution- and transform-based encoders, including [timm](https://github.com/huggingface/pytorch-image-models) support
2626
- Popular metrics and losses for training routines (Dice, Jaccard, Tversky, ...)
2727
- ONNX export and torch script/trace/compile friendly
@@ -123,6 +123,7 @@ Congratulations! You are done! Now you can train your model with your favorite f
123123
- DeepLabV3+ [[paper](https://arxiv.org/abs/1802.02611)] [[docs](https://smp.readthedocs.io/en/latest/models.html#id9)]
124124
- UPerNet [[paper](https://arxiv.org/abs/1807.10221)] [[docs](https://smp.readthedocs.io/en/latest/models.html#upernet)]
125125
- Segformer [[paper](https://arxiv.org/abs/2105.15203)] [[docs](https://smp.readthedocs.io/en/latest/models.html#segformer)]
126+
- DPT [[paper](https://arxiv.org/abs/2103.13413)] [[docs](https://smp.readthedocs.io/en/latest/models.html#dpt)]
126127

127128
### Encoders <a name="encoders"></a>
128129

0 commit comments

Comments
 (0)