qubvel-hf HF staff commited on
Commit
e67aa80
·
verified ·
1 Parent(s): 3a1c1c9

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
+ - mobilenet
11
+ languages:
12
+ - python
13
+ ---
14
+
15
+ # Model card for mobilenet_v2.
16
+
17
+ This repository contains the `imagenet` pre-trained weights for the `mobilenet_v2` 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(mobilenet_v2, 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://download.pytorch.org/models/mobilenet_v2-b0353104.pth
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:be3534eab2a7ad16d91fbd299afdf63e12dd2a20062577a1b63d91f9ef6d1510
3
+ size 14186200