qubvel-hf HF staff commited on
Commit
868ab6f
·
verified ·
1 Parent(s): 498e203

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
+ - mit
11
+ languages:
12
+ - python
13
+ ---
14
+
15
+ # Model card for mit_b2.
16
+
17
+ This repository contains the `imagenet` pre-trained weights for the `mit_b2` 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(mit_b2, 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://github.com/qubvel/segmentation_models.pytorch/releases/download/v0.0.2/mit_b2.pth
41
+
config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"input_space": "RGB", "input_size": [3, 224, 224], "input_range": [0, 1], "mean": [0.485, 0.456, 0.406], "std": [0.229, 0.224, 0.225]}
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ab18b54f81748427bae1183160822903470bcca0e3121944c0d185b4b1816a95
3
+ size 98867640