qubvel-hf HF staff commited on
Commit
0e3a35d
·
verified ·
1 Parent(s): c7f78e8

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. README.md +1 -1
  2. 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='swsl')
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}