qubvel-hf HF staff commited on
Commit
ddc3db8
·
verified ·
1 Parent(s): 9f6953f

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. README.md +41 -0
  2. config.json +1 -0
  3. model.safetensors +3 -0
README.md ADDED
@@ -0,0 +1,41 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ ---
3
+ library_name: segmentation-models-pytorch
4
+ license: other
5
+ pipeline_tag: image-classification
6
+ tags:
7
+ - segmentation-models-pytorch
8
+ - image-classification
9
+ - pytorch
10
+ - mobileone
11
+ languages:
12
+ - python
13
+ ---
14
+
15
+ # Model card for mobileone_s2.
16
+
17
+ This repository contains the `imagenet` pre-trained weights for the `mobileone_s2` model used as
18
+ encoder in the [segmentation-models-pytorch](https://github.com/qubvel-org/segmentation_models.pytorch) library.
19
+
20
+ ### Example usage:
21
+
22
+ 1. Install the library:
23
+
24
+ ```bash
25
+ pip install segmentation-models-pytorch
26
+ ```
27
+
28
+ 2. Use the encoder in your code:
29
+
30
+ ```python
31
+ import segmentation_models_pytorch as smp
32
+
33
+ model = smp.Unet(mobileone_s2, encoder_weights="imagenet")
34
+ ```
35
+
36
+ ### References
37
+
38
+ - Github: https://github.com/qubvel/segmentation_models.pytorch
39
+ - Docs: https://smp.readthedocs.io/en/latest/
40
+ - Original weights URL: https://docs-assets.developer.apple.com/ml-research/datasets/mobileone/mobileone_s2_unfused.pth.tar
41
+
config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"mean": [0.485, 0.456, 0.406], "std": [0.229, 0.224, 0.225], "input_space": "RGB", "input_range": [0, 1]}
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d7b3fc0a777bd04dd7a0ec25c7492709dc6a3b3ecd3ec0a3d875775294831e22
3
+ size 31942224