izou3 commited on
Commit
1684591
1 Parent(s): 82124db

Upload 2 files

Browse files
Files changed (2) hide show
  1. config.json +112 -0
  2. pytorch_model.bin +3 -0
config.json ADDED
@@ -0,0 +1,112 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "nvidia/mit-b3",
3
+ "architectures": [
4
+ "SegformerForSemanticSegmentation"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "classifier_dropout_prob": 0.1,
8
+ "decoder_hidden_size": 768,
9
+ "depths": [
10
+ 3,
11
+ 4,
12
+ 18,
13
+ 3
14
+ ],
15
+ "downsampling_rates": [
16
+ 1,
17
+ 4,
18
+ 8,
19
+ 16
20
+ ],
21
+ "drop_path_rate": 0.1,
22
+ "hidden_act": "gelu",
23
+ "hidden_dropout_prob": 0.0,
24
+ "hidden_sizes": [
25
+ 64,
26
+ 128,
27
+ 320,
28
+ 512
29
+ ],
30
+ "id2label": {
31
+ "0": "unlabeled",
32
+ "1": "Pizza",
33
+ "2": "Apple",
34
+ "3": "Brownie",
35
+ "4": "Lettuce",
36
+ "5": "Pasta",
37
+ "6": "Sweet Potato Fries",
38
+ "7": "Beans",
39
+ "8": "Beef",
40
+ "9": "Bread",
41
+ "10": "Cake",
42
+ "11": "Chicken",
43
+ "12": "Corn",
44
+ "13": "Green_bean",
45
+ "14": "Hot_dog_bun",
46
+ "15": "Pork",
47
+ "16": "Potato",
48
+ "17": "Rice",
49
+ "18": "Plate"
50
+ },
51
+ "image_size": 224,
52
+ "initializer_range": 0.02,
53
+ "label2id": {
54
+ "Apple": 2,
55
+ "Beans": 7,
56
+ "Beef": 8,
57
+ "Bread": 9,
58
+ "Brownie": 3,
59
+ "Cake": 10,
60
+ "Chicken": 11,
61
+ "Corn": 12,
62
+ "Green_bean": 13,
63
+ "Hot_dog_bun": 14,
64
+ "Lettuce": 4,
65
+ "Pasta": 5,
66
+ "Pizza": 1,
67
+ "Plate": 18,
68
+ "Pork": 15,
69
+ "Potato": 16,
70
+ "Rice": 17,
71
+ "Sweet Potato Fries": 6,
72
+ "unlabeled": 0
73
+ },
74
+ "layer_norm_eps": 1e-06,
75
+ "mlp_ratios": [
76
+ 4,
77
+ 4,
78
+ 4,
79
+ 4
80
+ ],
81
+ "model_type": "segformer",
82
+ "num_attention_heads": [
83
+ 1,
84
+ 2,
85
+ 5,
86
+ 8
87
+ ],
88
+ "num_channels": 3,
89
+ "num_encoder_blocks": 4,
90
+ "patch_sizes": [
91
+ 7,
92
+ 3,
93
+ 3,
94
+ 3
95
+ ],
96
+ "reshape_last_stage": true,
97
+ "semantic_loss_ignore_index": 255,
98
+ "sr_ratios": [
99
+ 8,
100
+ 4,
101
+ 2,
102
+ 1
103
+ ],
104
+ "strides": [
105
+ 4,
106
+ 2,
107
+ 2,
108
+ 2
109
+ ],
110
+ "torch_dtype": "float32",
111
+ "transformers_version": "4.25.1"
112
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f294cdff749a9e11d51a5c227b62ea03977d5a99a64cd7507549c91fd2ce4cc8
3
+ size 189173113