hf-transformers-bot commited on
Commit
49b0578
·
verified ·
1 Parent(s): 4e13f6c

Update tiny models for MaskFormerForInstanceSegmentation

Browse files
Files changed (3) hide show
  1. config.json +149 -0
  2. model.safetensors +3 -0
  3. preprocessor_config.json +26 -0
config.json ADDED
@@ -0,0 +1,149 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "MaskFormerForInstanceSegmentation"
4
+ ],
5
+ "backbone": null,
6
+ "backbone_config": {
7
+ "attention_probs_dropout_prob": 0.0,
8
+ "depths": [
9
+ 1,
10
+ 1,
11
+ 1,
12
+ 1
13
+ ],
14
+ "drop_path_rate": 0.1,
15
+ "embed_dim": 16,
16
+ "encoder_stride": 32,
17
+ "hidden_act": "gelu",
18
+ "hidden_dropout_prob": 0.0,
19
+ "hidden_size": 32,
20
+ "image_size": 224,
21
+ "initializer_range": 0.02,
22
+ "layer_norm_eps": 1e-05,
23
+ "mlp_ratio": 4.0,
24
+ "model_type": "maskformer-swin",
25
+ "num_channels": 3,
26
+ "num_heads": [
27
+ 1,
28
+ 1,
29
+ 2,
30
+ 2
31
+ ],
32
+ "num_layers": 4,
33
+ "out_features": [
34
+ "stage1",
35
+ "stage2",
36
+ "stage3",
37
+ "stage4"
38
+ ],
39
+ "out_indices": [
40
+ 1,
41
+ 2,
42
+ 3,
43
+ 4
44
+ ],
45
+ "patch_size": 4,
46
+ "qkv_bias": true,
47
+ "stage_names": [
48
+ "stem",
49
+ "stage1",
50
+ "stage2",
51
+ "stage3",
52
+ "stage4"
53
+ ],
54
+ "use_absolute_embeddings": false,
55
+ "window_size": 7
56
+ },
57
+ "cross_entropy_weight": 1.0,
58
+ "decoder_config": {
59
+ "activation_dropout": 0.0,
60
+ "activation_function": "relu",
61
+ "attention_dropout": 0.0,
62
+ "auxiliary_loss": false,
63
+ "backbone_config": {
64
+ "_name_or_path": "",
65
+ "architectures": null,
66
+ "backbone": "resnet50",
67
+ "chunk_size_feed_forward": 0,
68
+ "dtype": null,
69
+ "features_only": true,
70
+ "freeze_batch_norm_2d": false,
71
+ "id2label": {
72
+ "0": "LABEL_0",
73
+ "1": "LABEL_1"
74
+ },
75
+ "is_encoder_decoder": false,
76
+ "label2id": {
77
+ "LABEL_0": 0,
78
+ "LABEL_1": 1
79
+ },
80
+ "model_type": "timm_backbone",
81
+ "num_channels": 3,
82
+ "out_features": null,
83
+ "out_indices": [
84
+ 1,
85
+ 2,
86
+ 3,
87
+ 4
88
+ ],
89
+ "output_attentions": false,
90
+ "output_hidden_states": false,
91
+ "output_stride": null,
92
+ "problem_type": null,
93
+ "return_dict": true,
94
+ "use_pretrained_backbone": false
95
+ },
96
+ "bbox_cost": 5,
97
+ "bbox_loss_coefficient": 5,
98
+ "class_cost": 1,
99
+ "d_model": 32,
100
+ "decoder_attention_heads": 2,
101
+ "decoder_ffn_dim": 64,
102
+ "decoder_layerdrop": 0.0,
103
+ "decoder_layers": 2,
104
+ "dice_loss_coefficient": 1,
105
+ "dilation": false,
106
+ "dropout": 0.1,
107
+ "encoder_attention_heads": 2,
108
+ "encoder_ffn_dim": 64,
109
+ "encoder_layerdrop": 0.0,
110
+ "encoder_layers": 2,
111
+ "eos_coefficient": 0.1,
112
+ "giou_cost": 2,
113
+ "giou_loss_coefficient": 2,
114
+ "init_std": 0.02,
115
+ "init_xavier_std": 1.0,
116
+ "mask_loss_coefficient": 1,
117
+ "model_type": "detr",
118
+ "num_channels": 3,
119
+ "num_queries": 10,
120
+ "position_embedding_type": "sine"
121
+ },
122
+ "dice_weight": 1.0,
123
+ "dtype": "float32",
124
+ "fpn_feature_size": 32,
125
+ "id2label": {
126
+ "0": "LABEL_0",
127
+ "1": "LABEL_1",
128
+ "2": "LABEL_2",
129
+ "3": "LABEL_3"
130
+ },
131
+ "init_std": 0.02,
132
+ "init_xavier_std": 1.0,
133
+ "label2id": {
134
+ "LABEL_0": 0,
135
+ "LABEL_1": 1,
136
+ "LABEL_2": 2,
137
+ "LABEL_3": 3
138
+ },
139
+ "mask_feature_size": 32,
140
+ "mask_weight": 20.0,
141
+ "model_type": "maskformer",
142
+ "no_object_weight": 0.1,
143
+ "num_attention_heads": 2,
144
+ "num_channels": 3,
145
+ "num_hidden_layers": 2,
146
+ "output_auxiliary_logits": null,
147
+ "transformers_version": "5.16.0.dev0",
148
+ "use_auxiliary_loss": false
149
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ee5d40f6c54c2157aeb603f4f454983172459b1dd5bc8400ee1894e1e6ad4a7e
3
+ size 1790120
preprocessor_config.json ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_normalize": true,
3
+ "do_reduce_labels": false,
4
+ "do_rescale": true,
5
+ "do_resize": true,
6
+ "ignore_index": 255,
7
+ "image_mean": [
8
+ 0.48500001430511475,
9
+ 0.4560000002384186,
10
+ 0.4059999883174896
11
+ ],
12
+ "image_processor_type": "MaskFormerImageProcessor",
13
+ "image_std": [
14
+ 0.2290000021457672,
15
+ 0.2239999920129776,
16
+ 0.22499999403953552
17
+ ],
18
+ "resample": 2,
19
+ "rescale_factor": 0.00392156862745098,
20
+ "size": {
21
+ "longest_edge": 2560,
22
+ "shortest_edge": 640
23
+ },
24
+ "size_divisibility": 32,
25
+ "size_divisor": 32
26
+ }