vikp commited on
Commit
a386f21
1 Parent(s): 4636aee

Upload SegformerForRegressionMask

Browse files
Files changed (2) hide show
  1. config.json +80 -0
  2. model.safetensors +3 -0
config.json ADDED
@@ -0,0 +1,80 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "line_detector_math",
3
+ "architectures": [
4
+ "SegformerForRegressionMask"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "classifier_dropout_prob": 0.1,
8
+ "decoder_hidden_size": 768,
9
+ "decoder_layer_hidden_size": 192,
10
+ "decoder_upsample_rate": 2,
11
+ "depths": [
12
+ 3,
13
+ 4,
14
+ 9,
15
+ 3
16
+ ],
17
+ "downsampling_rates": [
18
+ 1,
19
+ 4,
20
+ 8,
21
+ 16
22
+ ],
23
+ "drop_path_rate": 0.1,
24
+ "hidden_act": "gelu",
25
+ "hidden_dropout_prob": 0.0,
26
+ "hidden_sizes": [
27
+ 64,
28
+ 128,
29
+ 320,
30
+ 512
31
+ ],
32
+ "id2label": {
33
+ "0": "blank",
34
+ "1": "text"
35
+ },
36
+ "image_size": 224,
37
+ "initializer_range": 0.02,
38
+ "label2id": {
39
+ "blank": 0,
40
+ "text": 1
41
+ },
42
+ "layer_norm_eps": 1e-06,
43
+ "mlp_ratios": [
44
+ 4,
45
+ 4,
46
+ 4,
47
+ 4
48
+ ],
49
+ "model_type": "segformer",
50
+ "num_attention_heads": [
51
+ 1,
52
+ 2,
53
+ 5,
54
+ 8
55
+ ],
56
+ "num_channels": 3,
57
+ "num_encoder_blocks": 4,
58
+ "patch_sizes": [
59
+ 7,
60
+ 3,
61
+ 3,
62
+ 3
63
+ ],
64
+ "reshape_last_stage": true,
65
+ "semantic_loss_ignore_index": -1,
66
+ "sr_ratios": [
67
+ 8,
68
+ 4,
69
+ 2,
70
+ 1
71
+ ],
72
+ "strides": [
73
+ 4,
74
+ 2,
75
+ 2,
76
+ 2
77
+ ],
78
+ "torch_dtype": "float32",
79
+ "transformers_version": "4.36.2"
80
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:32dd53dd16336139fe8ed1d1324483ec69495247d477c4fc89fa3cf8c9e155ac
3
+ size 119881240