Skip to content

Commit 03be12a

Browse files
authored
Update README.md
1 parent 1d0a657 commit 03be12a

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

README.md

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,10 +5,10 @@ Segmentation models is python library with Neural Networks for Image Segmentatio
55

66
The main features of this library are:
77

8-
- High level API (just two lines to create NN)
8+
- High level API (just two lines to create neural network)
99
- 4 models architectures for binary and multi class segmentation (including legendary Unet)
1010
- 30 available encoders for each architecture
11-
- All backbones have pre-trained weights for faster and better convergence
11+
- All encoders have pre-trained weights for faster and better convergence
1212

1313
### Table of content
1414
1. [Quick start](#start)
@@ -45,6 +45,8 @@ from segmentation_models_pytorch.encoders import get_preprocessing_fn
4545
preprocess_input = get_preprocessing_fn('renset18', pretrained='imagenet')
4646
```
4747

48+
**Model training example** on CamVid dataset [here](https://github.com/qubvel/segmentation_models.pytorch/blob/master/examples/cars%20segmentation%20(camvid).ipynb).
49+
4850
### Models <a name="models"></a>
4951

5052
#### Architectires <a name="architectires"></a>

0 commit comments

Comments
 (0)