Image Segmentation
Transformers
Safetensors
segformer
Zijian-Wang commited on
Commit
03e3e23
·
verified ·
1 Parent(s): 89fe9d2

Upload folder using huggingface_hub

Browse files
Files changed (4) hide show
  1. class.json +22 -0
  2. config.json +81 -0
  3. model.safetensors +3 -0
  4. preprocessor_config.json +23 -0
class.json ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "id": 0,
4
+ "name": "background",
5
+ "color": [0, 0, 0]
6
+ },
7
+ {
8
+ "id": 1,
9
+ "name": "head",
10
+ "color": [50, 255, 132]
11
+ },
12
+ {
13
+ "id": 2,
14
+ "name": "stem",
15
+ "color": [50, 132, 255]
16
+ },
17
+ {
18
+ "id": 3,
19
+ "name": "leaf",
20
+ "color": [214, 255, 50]
21
+ }
22
+ ]
config.json ADDED
@@ -0,0 +1,81 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "SegformerForSemanticSegmentation"
4
+ ],
5
+ "attention_probs_dropout_prob": 0.0,
6
+ "classifier_dropout_prob": 0.1,
7
+ "decoder_hidden_size": 256,
8
+ "depths": [
9
+ 2,
10
+ 2,
11
+ 2,
12
+ 2
13
+ ],
14
+ "downsampling_rates": [
15
+ 1,
16
+ 4,
17
+ 8,
18
+ 16
19
+ ],
20
+ "drop_path_rate": 0.1,
21
+ "hidden_act": "gelu",
22
+ "hidden_dropout_prob": 0.0,
23
+ "hidden_sizes": [
24
+ 64,
25
+ 128,
26
+ 320,
27
+ 512
28
+ ],
29
+ "id2label": {
30
+ "0": "background",
31
+ "1": "head",
32
+ "2": "stem",
33
+ "3": "leaf"
34
+ },
35
+ "image_size": 224,
36
+ "initializer_range": 0.02,
37
+ "label2id": {
38
+ "background": 0,
39
+ "head": 1,
40
+ "leaf": 3,
41
+ "stem": 2
42
+ },
43
+ "layer_norm_eps": 1e-06,
44
+ "mlp_ratios": [
45
+ 4,
46
+ 4,
47
+ 4,
48
+ 4
49
+ ],
50
+ "model_type": "segformer",
51
+ "num_attention_heads": [
52
+ 1,
53
+ 2,
54
+ 5,
55
+ 8
56
+ ],
57
+ "num_channels": 3,
58
+ "num_encoder_blocks": 4,
59
+ "patch_sizes": [
60
+ 7,
61
+ 3,
62
+ 3,
63
+ 3
64
+ ],
65
+ "reshape_last_stage": true,
66
+ "semantic_loss_ignore_index": 255,
67
+ "sr_ratios": [
68
+ 8,
69
+ 4,
70
+ 2,
71
+ 1
72
+ ],
73
+ "strides": [
74
+ 4,
75
+ 2,
76
+ 2,
77
+ 2
78
+ ],
79
+ "torch_dtype": "float32",
80
+ "transformers_version": "4.55.2"
81
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6d803e340ab0f99bb1e00d78faec8e16bef145a2eec08dc315ac2bb2fc30e95a
3
+ size 54739432
preprocessor_config.json ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_normalize": true,
3
+ "do_reduce_labels": false,
4
+ "do_rescale": true,
5
+ "do_resize": true,
6
+ "image_mean": [
7
+ 0.485,
8
+ 0.456,
9
+ 0.406
10
+ ],
11
+ "image_processor_type": "SegformerImageProcessor",
12
+ "image_std": [
13
+ 0.229,
14
+ 0.224,
15
+ 0.225
16
+ ],
17
+ "resample": 2,
18
+ "rescale_factor": 0.00392156862745098,
19
+ "size": {
20
+ "height": 512,
21
+ "width": 512
22
+ }
23
+ }