qubvel-hf HF staff commited on
Commit
f52815c
·
verified ·
1 Parent(s): afc7e6e

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_s0.
16
+
17
+ This repository contains the `imagenet` pre-trained weights for the `mobileone_s0` 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_s0, 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_s0_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:6a68e86f917c53ec6f88824109608ba686698b1c1b7a1b8b84082573e60e5694
3
+ size 21657776