qubvel-hf HF staff commited on
Commit
bea8b0f
·
verified ·
1 Parent(s): 6b7fc7c

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
+ - efficientnet
11
+ languages:
12
+ - python
13
+ ---
14
+
15
+ # Model card for timm-efficientnet-b0.
16
+
17
+ This repository contains the `noisy-student` pre-trained weights for the `timm-efficientnet-b0` 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(timm-efficientnet-b0, encoder_weights="noisy-student")
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/rwightman/pytorch-image-models/releases/download/v0.1-weights/tf_efficientnet_b0_ns-c0e6a31c.pth
41
+
config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"mean": [0.485, 0.456, 0.406], "std": [0.229, 0.224, 0.225], "input_range": [0, 1], "input_space": "RGB"}
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ca6fd4de70b4777557e7deca534935b5db289946d55c98c0b589c9c399771510
3
+ size 21355344