File tree Expand file tree Collapse file tree 2 files changed +9
-1
lines changed Expand file tree Collapse file tree 2 files changed +9
-1
lines changed Original file line number Diff line number Diff line change @@ -19,7 +19,7 @@ Segmentation based on [PyTorch](https://pytorch.org/).**
19
19
The main features of this library are:
20
20
21
21
- 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)
23
23
- 124 available encoders (and 500+ encoders from [ timm] ( https://github.com/rwightman/pytorch-image-models ) )
24
24
- All encoders have pre-trained weights for faster and better convergence
25
25
- 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
94
94
- PAN [[ paper] ( https://arxiv.org/abs/1805.10180 )] [[ docs] ( https://smp.readthedocs.io/en/latest/models.html#pan )]
95
95
- DeepLabV3 [[ paper] ( https://arxiv.org/abs/1706.05587 )] [[ docs] ( https://smp.readthedocs.io/en/latest/models.html#deeplabv3 )]
96
96
- 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 )]
97
98
98
99
#### Encoders <a name =" encoders " ></a >
99
100
Original file line number Diff line number Diff line change 66
66
PAN
67
67
~~~
68
68
.. autoclass :: segmentation_models_pytorch.PAN
69
+
70
+
71
+ .. _upernet :
72
+
73
+ UPerNet
74
+ ~~~
75
+ .. autoclass :: segmentation_models_pytorch.UPerNet
You can’t perform that action at this time.
0 commit comments