Skip to content

Commit 5e37177

Browse files
committed
update readme and doc
1 parent 748070e commit 5e37177

File tree

2 files changed

+9
-1
lines changed

2 files changed

+9
-1
lines changed

README.md

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

2121
- High-level API (just two lines to create a neural network)
22-
- 9 models architectures for binary and multi class segmentation (including legendary Unet)
22+
- 10 models architectures for binary and multi class segmentation (including legendary Unet)
2323
- 124 available encoders (and 500+ encoders from [timm](https://github.com/rwightman/pytorch-image-models))
2424
- All encoders have pre-trained weights for faster and better convergence
2525
- Popular metrics and losses for training routines
@@ -94,6 +94,7 @@ Congratulations! You are done! Now you can train your model with your favorite f
9494
- PAN [[paper](https://arxiv.org/abs/1805.10180)] [[docs](https://smp.readthedocs.io/en/latest/models.html#pan)]
9595
- DeepLabV3 [[paper](https://arxiv.org/abs/1706.05587)] [[docs](https://smp.readthedocs.io/en/latest/models.html#deeplabv3)]
9696
- DeepLabV3+ [[paper](https://arxiv.org/abs/1802.02611)] [[docs](https://smp.readthedocs.io/en/latest/models.html#id9)]
97+
- UPerNet [[paper](https://arxiv.org/abs/1807.10221)] [[docs](https://smp.readthedocs.io/en/latest/models.html#upernet)]
9798

9899
#### Encoders <a name="encoders"></a>
99100

docs/models.rst

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -66,3 +66,10 @@ MAnet
6666
PAN
6767
~~~
6868
.. autoclass:: segmentation_models_pytorch.PAN
69+
70+
71+
.. _upernet:
72+
73+
UPerNet
74+
~~~
75+
.. autoclass:: segmentation_models_pytorch.UPerNet

0 commit comments

Comments
 (0)