Skip to content

Commit 8f10772

Browse files
authored
Update README.md
1 parent a41f650 commit 8f10772

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

README.md

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -12,12 +12,13 @@ The main features of this library are:
1212

1313
### Table of content
1414
1. [Quick start](#start)
15-
2. [Models](#models)
15+
2. [Examples](#examples)
16+
3. [Models](#models)
1617
1. [Architectires](#architectires)
1718
2. [Encoders](#encoders)
1819
3. [Pretrained weights](#weights)
19-
3. [Installation](#installation)
20-
4. [License](#license)
20+
4. [Installation](#installation)
21+
5. [License](#license)
2122

2223
### Quick start <a name="start"></a>
2324
Since the library is built on the PyTorch framework, created segmentation model is just a PyTorch nn.Module, which can be created as easy as:
@@ -44,8 +45,9 @@ from segmentation_models_pytorch.encoders import get_preprocessing_fn
4445

4546
preprocess_input = get_preprocessing_fn('renset18', pretrained='imagenet')
4647
```
47-
48-
**Model training example** on CamVid dataset [here](https://github.com/qubvel/segmentation_models.pytorch/blob/master/examples/cars%20segmentation%20(camvid).ipynb).
48+
### Examples <a name="examples"></a>
49+
- Training model for cars segmentation on CamVid dataset [here](https://github.com/qubvel/segmentation_models.pytorch/blob/master/examples/cars%20segmentation%20(camvid).ipynb).
50+
- Training model with [Catalyst](https://github.com/catalyst-team/catalyst) (high-level framework for PyTorch) - [here](https://colab.research.google.com/gist/Scitator/e3fd90eec05162e16b476de832500576/cars-segmentation-camvid.ipynb).
4951

5052
### Models <a name="models"></a>
5153

0 commit comments

Comments
 (0)