Skip to content

Commit 9c7eedf

Browse files
authored
Update README.md
1 parent 97181b9 commit 9c7eedf

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

README.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@ model = smp.Unet('resnet34', classes=3, activation='softmax')
3232
As all models have pretrained weights, so you have to prepare your data the same way as during weights pretraining:
3333
```python
3434
from segmentation_models_pytorch.encoders import get_preprocessing_fn
35+
3536
preprocess_input = get_preprocessing_fn('renset18', pretrained='imagenet')
3637
```
3738

0 commit comments

Comments
 (0)