File tree Expand file tree Collapse file tree 1 file changed +2
-1
lines changed Expand file tree Collapse file tree 1 file changed +2
-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
- - 10 models architectures for binary and multi class segmentation (including legendary Unet)
22
+ - 11 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
@@ -95,6 +95,7 @@ Congratulations! You are done! Now you can train your model with your favorite f
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
97
- UPerNet [[ paper] ( https://arxiv.org/abs/1807.10221 )] [[ docs] ( https://smp.readthedocs.io/en/latest/models.html#upernet )]
98
+ - Segformer [[ paper] ( https://arxiv.org/abs/2105.15203 )] [[ docs] ( https://smp.readthedocs.io/en/latest/models.html#segformer )]
98
99
99
100
#### Encoders <a name =" encoders " ></a >
100
101
You can’t perform that action at this time.
0 commit comments