Upload folder using huggingface_hub
Browse files- README.md +1 -1
- config.json +1 -0
README.md
CHANGED
@@ -30,7 +30,7 @@ pip install segmentation-models-pytorch
|
|
30 |
```python
|
31 |
import segmentation_models_pytorch as smp
|
32 |
|
33 |
-
model = smp.Unet(resnet18, encoder_weights=
|
34 |
```
|
35 |
|
36 |
### References
|
|
|
30 |
```python
|
31 |
import segmentation_models_pytorch as smp
|
32 |
|
33 |
+
model = smp.Unet(resnet18, encoder_weights="swsl")
|
34 |
```
|
35 |
|
36 |
### References
|
config.json
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
{"input_size": [3, 224, 224], "input_range": [0, 1], "mean": [0.485, 0.456, 0.406], "std": [0.229, 0.224, 0.225], "num_classes": 1000}
|