Update README.md
Browse files
README.md
CHANGED
|
@@ -26,6 +26,9 @@ import segmentation_models_pytorch as smp
|
|
| 26 |
model = smp.from_pretrained("smp-hub/segformer-b2-1024x1024-city-160k")
|
| 27 |
```
|
| 28 |
|
|
|
|
|
|
|
|
|
|
| 29 |
## Model init parameters
|
| 30 |
```python
|
| 31 |
model_init_params = {
|
|
|
|
| 26 |
model = smp.from_pretrained("smp-hub/segformer-b2-1024x1024-city-160k")
|
| 27 |
```
|
| 28 |
|
| 29 |
+
Inference Example:
|
| 30 |
+
- https://github.com/qubvel-org/segmentation_models.pytorch/blob/main/examples/segformer_inference_pretrained.ipynb
|
| 31 |
+
|
| 32 |
## Model init parameters
|
| 33 |
```python
|
| 34 |
model_init_params = {
|