Instructions to use obrito/my-segmentation-model with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use obrito/my-segmentation-model with Transformers:
# Load model directly from transformers import AutoImageProcessor, SegformerForSemanticSegmentation processor = AutoImageProcessor.from_pretrained("obrito/my-segmentation-model") model = SegformerForSemanticSegmentation.from_pretrained("obrito/my-segmentation-model") - Notebooks
- Google Colab
- Kaggle
Upload SegformerForSemanticSegmentation
Browse files- config.json +5 -5
- pytorch_model.bin +1 -1
config.json
CHANGED
|
@@ -37,11 +37,11 @@
|
|
| 37 |
"image_size": 224,
|
| 38 |
"initializer_range": 0.02,
|
| 39 |
"label2id": {
|
| 40 |
-
"background":
|
| 41 |
-
"human":
|
| 42 |
-
"road":
|
| 43 |
-
"traffic lights":
|
| 44 |
-
"vehicles":
|
| 45 |
},
|
| 46 |
"layer_norm_eps": 1e-06,
|
| 47 |
"mlp_ratios": [
|
|
|
|
| 37 |
"image_size": 224,
|
| 38 |
"initializer_range": 0.02,
|
| 39 |
"label2id": {
|
| 40 |
+
"background": 4,
|
| 41 |
+
"human": 1,
|
| 42 |
+
"road": 0,
|
| 43 |
+
"traffic lights": 3,
|
| 44 |
+
"vehicles": 2
|
| 45 |
},
|
| 46 |
"layer_norm_eps": 1e-06,
|
| 47 |
"mlp_ratios": [
|
pytorch_model.bin
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 14934861
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:04f76b98691d02c4d59d1404da961482d789ab98a1b6d4ade4cb7cce1cd81f43
|
| 3 |
size 14934861
|