Update segmentation_dataset.py

YknZhu · web-flow · commit e225b20ab62f · 2018-05-18T19:13:47.000-07:00
Update ADE20K dataset info.
diff --git a/research/deeplab/datasets/segmentation_dataset.py b/research/deeplab/datasets/segmentation_dataset.py
@@ -103,8 +103,8 @@
         'train': 20210,  # num of samples in images/training
         'val': 2000,  # num of samples in images/validation
     },
-    num_classes=150,
-    ignore_label=255,
+    num_classes=151,
+    ignore_label=0,
 )
 
 

Original file line number	Diff line number	Diff line change
`@@ -103,8 +103,8 @@`
`103`	`103`	`'train': 20210, # num of samples in images/training`
`104`	`104`	`'val': 2000, # num of samples in images/validation`
`105`	`105`	`},`
`106`		`- num_classes=150,`
`107`		`- ignore_label=255,`
	`106`	`+ num_classes=151,`
	`107`	`+ ignore_label=0,`
`108`	`108`	`)`
`109`	`109`
`110`	`110`