Image Segmentation
Transformers
PyTorch
upernet
mccaly commited on
Commit
51a392d
·
1 Parent(s): ca97bc4

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +2 -20
config.json CHANGED
@@ -1,5 +1,4 @@
1
  {
2
-
3
  "model_type": "unet",
4
  "model_config": {
5
  "encoder": {
@@ -10,8 +9,7 @@
10
  "type": "unet",
11
  "num_classes": 104
12
  }
13
- }
14
- }
15
  "norm_cfg": {
16
  "type": "SyncBN",
17
  "requires_grad": true
@@ -213,20 +211,4 @@
213
  "type": "DefaultFormatBundle"
214
  },
215
  {
216
- "type": "Collect",
217
- "keys": ["img", "gt_semantic_seg"]
218
- }
219
- ]
220
- },
221
- "val": {
222
- "type": "CustomDataset",
223
- "data_root": "./data/FoodSeg103/Images/",
224
- "img_dir": "img_dir/test",
225
- "ann_dir": "ann_dir/test",
226
- "pipeline": [
227
- {
228
- "type": "LoadImageFromFile"
229
- },
230
- {
231
- "type": "MultiScaleFlipAug",
232
- "img_scale": [2048, 1024],
 
1
  {
 
2
  "model_type": "unet",
3
  "model_config": {
4
  "encoder": {
 
9
  "type": "unet",
10
  "num_classes": 104
11
  }
12
+ },
 
13
  "norm_cfg": {
14
  "type": "SyncBN",
15
  "requires_grad": true
 
211
  "type": "DefaultFormatBundle"
212
  },
213
  {
214
+ "type": "Collect",