Patnev71 commited on
Commit
ecc4fa6
·
verified ·
1 Parent(s): 0f63450

Upload SegformerForSemanticSegmentation

Browse files
Files changed (2) hide show
  1. config.json +18 -20
  2. model.safetensors +2 -2
config.json CHANGED
@@ -28,30 +28,28 @@
28
  256
29
  ],
30
  "id2label": {
31
- "0": "undefined",
32
- "1": "background",
33
- "2": "living",
34
- "3": "kitchen",
35
- "4": "circulation",
36
- "5": "bathroom",
37
- "6": "utility",
38
- "7": "bedroom",
39
- "8": "storage",
40
- "9": "garage"
41
  },
42
  "image_size": 224,
43
  "initializer_range": 0.02,
44
  "label2id": {
45
- "background": 1,
46
- "bathroom": 5,
47
- "bedroom": 7,
48
- "circulation": 4,
49
- "garage": 9,
50
- "kitchen": 3,
51
- "living": 2,
52
- "storage": 8,
53
- "undefined": 0,
54
- "utility": 6
55
  },
56
  "layer_norm_eps": 1e-06,
57
  "mlp_ratios": [
 
28
  256
29
  ],
30
  "id2label": {
31
+ "0": "background",
32
+ "1": "living",
33
+ "2": "kitchen",
34
+ "3": "circulation",
35
+ "4": "bathroom",
36
+ "5": "utility",
37
+ "6": "bedroom",
38
+ "7": "storage",
39
+ "8": "garage"
 
40
  },
41
  "image_size": 224,
42
  "initializer_range": 0.02,
43
  "label2id": {
44
+ "background": 0,
45
+ "bathroom": 4,
46
+ "bedroom": 6,
47
+ "circulation": 3,
48
+ "garage": 8,
49
+ "kitchen": 2,
50
+ "living": 1,
51
+ "storage": 7,
52
+ "utility": 5
 
53
  },
54
  "layer_norm_eps": 1e-06,
55
  "mlp_ratios": [
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3baba40bbdb219c192bcbba6a44815aa5667a02cba9efa6f723b110afadcacef
3
- size 14893008
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dfba35a1c98451422bd147d824ceca14894209385ca82c6e41ed8a81d8998cf6
3
+ size 14891980