Hugging Face's logo

keras
/
segformer_b1_ade20k_512

KerasHub
Model card Files Community
Divyasreepat commited on
Commit
e2c4ee8
•
1 Parent(s): 3fb0cd1

Upload folder using huggingface_hub

Browse files
Files changed (7) hide show
  1. README.md +13 -0
  2. config.json +62 -0
  3. metadata.json +6 -0
  4. model.weights.h5 +3 -0
  5. preprocessor.json +18 -0
  6. task.json +89 -0
  7. task.weights.h5 +3 -0
README.md ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: keras-hub
3
+ ---
4
+ This is a [`SegFormer` model](https://keras.io/api/keras_hub/models/seg_former) uploaded using the KerasHub library and can be used with JAX, TensorFlow, and PyTorch backends.
5
+ This model is related to a `ImageSegmenter` task.
6
+
7
+ Model config:
8
+ * **name:** seg_former_backbone
9
+ * **trainable:** True
10
+ * **projection_filters:** 256
11
+ * **image_encoder:** {'module': 'keras_hub.src.models.mit.mit_backbone', 'class_name': 'MiTBackbone', 'config': {'name': 'mi_t_backbone', 'trainable': True, 'layerwise_depths': [2, 2, 2, 2], 'hidden_dims': [64, 128, 320, 512], 'image_shape': [512, 512, 3], 'num_layers': 4, 'layerwise_num_heads': [1, 2, 5, 8], 'layerwise_sr_ratios': [8, 4, 2, 1], 'max_drop_path_rate': 0.1, 'layerwise_patch_sizes': [7, 3, 3, 3], 'layerwise_strides': [4, 2, 2, 2]}, 'registered_name': 'keras_hub>MiTBackbone'}
12
+
13
+ This model card has been generated automatically and should be completed by the model author. See [Model Cards documentation](https://huggingface.co/docs/hub/model-cards) for more information.
config.json ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "module": "keras_hub.src.models.segformer.segformer_backbone",
3
+ "class_name": "SegFormerBackbone",
4
+ "config": {
5
+ "name": "seg_former_backbone",
6
+ "trainable": true,
7
+ "projection_filters": 256,
8
+ "image_encoder": {
9
+ "module": "keras_hub.src.models.mit.mit_backbone",
10
+ "class_name": "MiTBackbone",
11
+ "config": {
12
+ "name": "mi_t_backbone",
13
+ "trainable": true,
14
+ "layerwise_depths": [
15
+ 2,
16
+ 2,
17
+ 2,
18
+ 2
19
+ ],
20
+ "hidden_dims": [
21
+ 64,
22
+ 128,
23
+ 320,
24
+ 512
25
+ ],
26
+ "image_shape": [
27
+ 512,
28
+ 512,
29
+ 3
30
+ ],
31
+ "num_layers": 4,
32
+ "layerwise_num_heads": [
33
+ 1,
34
+ 2,
35
+ 5,
36
+ 8
37
+ ],
38
+ "layerwise_sr_ratios": [
39
+ 8,
40
+ 4,
41
+ 2,
42
+ 1
43
+ ],
44
+ "max_drop_path_rate": 0.1,
45
+ "layerwise_patch_sizes": [
46
+ 7,
47
+ 3,
48
+ 3,
49
+ 3
50
+ ],
51
+ "layerwise_strides": [
52
+ 4,
53
+ 2,
54
+ 2,
55
+ 2
56
+ ]
57
+ },
58
+ "registered_name": "keras_hub>MiTBackbone"
59
+ }
60
+ },
61
+ "registered_name": "keras_hub>SegFormerBackbone"
62
+ }
metadata.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "keras_version": "3.6.0",
3
+ "keras_hub_version": "0.17.0.dev0",
4
+ "parameter_count": 13677760,
5
+ "date_saved": "2024-10-30@15:27:41"
6
+ }
model.weights.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4e49e671372c87aa01cb2901f53fbff8d889976ee0a5757ac4f61d1a37c0a165
3
+ size 55144784
preprocessor.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "module": "keras_hub.src.models.segformer.segformer_image_segmenter_preprocessor",
3
+ "class_name": "SegFormerImageSegmenterPreprocessor",
4
+ "config": {
5
+ "name": "seg_former_image_segmenter_preprocessor",
6
+ "trainable": true,
7
+ "dtype": {
8
+ "module": "keras",
9
+ "class_name": "DTypePolicy",
10
+ "config": {
11
+ "name": "float32"
12
+ },
13
+ "registered_name": null
14
+ },
15
+ "config_file": "preprocessor.json"
16
+ },
17
+ "registered_name": "keras_hub>SegFormerImageSegmenterPreprocessor"
18
+ }
task.json ADDED
@@ -0,0 +1,89 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "module": "keras_hub.src.models.segformer.segformer_image_segmenter",
3
+ "class_name": "SegFormerImageSegmenter",
4
+ "config": {
5
+ "backbone": {
6
+ "module": "keras_hub.src.models.segformer.segformer_backbone",
7
+ "class_name": "SegFormerBackbone",
8
+ "config": {
9
+ "name": "seg_former_backbone",
10
+ "trainable": true,
11
+ "projection_filters": 256,
12
+ "image_encoder": {
13
+ "module": "keras_hub.src.models.mit.mit_backbone",
14
+ "class_name": "MiTBackbone",
15
+ "config": {
16
+ "name": "mi_t_backbone",
17
+ "trainable": true,
18
+ "layerwise_depths": [
19
+ 2,
20
+ 2,
21
+ 2,
22
+ 2
23
+ ],
24
+ "hidden_dims": [
25
+ 64,
26
+ 128,
27
+ 320,
28
+ 512
29
+ ],
30
+ "image_shape": [
31
+ 512,
32
+ 512,
33
+ 3
34
+ ],
35
+ "num_layers": 4,
36
+ "layerwise_num_heads": [
37
+ 1,
38
+ 2,
39
+ 5,
40
+ 8
41
+ ],
42
+ "layerwise_sr_ratios": [
43
+ 8,
44
+ 4,
45
+ 2,
46
+ 1
47
+ ],
48
+ "max_drop_path_rate": 0.1,
49
+ "layerwise_patch_sizes": [
50
+ 7,
51
+ 3,
52
+ 3,
53
+ 3
54
+ ],
55
+ "layerwise_strides": [
56
+ 4,
57
+ 2,
58
+ 2,
59
+ 2
60
+ ]
61
+ },
62
+ "registered_name": "keras_hub>MiTBackbone"
63
+ }
64
+ },
65
+ "registered_name": "keras_hub>SegFormerBackbone"
66
+ },
67
+ "preprocessor": {
68
+ "module": "keras_hub.src.models.segformer.segformer_image_segmenter_preprocessor",
69
+ "class_name": "SegFormerImageSegmenterPreprocessor",
70
+ "config": {
71
+ "name": "seg_former_image_segmenter_preprocessor",
72
+ "trainable": true,
73
+ "dtype": {
74
+ "module": "keras",
75
+ "class_name": "DTypePolicy",
76
+ "config": {
77
+ "name": "float32"
78
+ },
79
+ "registered_name": null
80
+ },
81
+ "config_file": "preprocessor.json"
82
+ },
83
+ "registered_name": "keras_hub>SegFormerImageSegmenterPreprocessor"
84
+ },
85
+ "name": "seg_former_image_segmenter",
86
+ "num_classes": 150
87
+ },
88
+ "registered_name": "keras_hub>SegFormerImageSegmenter"
89
+ }
task.weights.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2946e0da251cc15eecf5e344c8eeb32529a8289e95d137550fd5e3ab47ec6528
3
+ size 55311912