Update segmentation_dataset.py

Update ADE20K dataset info.

Update segmentation_dataset.py
Update ADE20K dataset info.
e225b20a · Yukun Zhu · GitHub · 76d494bd · e225b20a
Unverified Commit e225b20a authored May 18, 2018 by Yukun Zhu Committed by GitHub May 18, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

research/deeplab/datasets/segmentation_dataset.py research/deeplab/datasets/segmentation_dataset.py +2 -2

No files found.
--- a/research/deeplab/datasets/segmentation_dataset.py
+++ b/research/deeplab/datasets/segmentation_dataset.py
@@ -103,8 +103,8 @@ _ADE20K_INFORMATION = DatasetDescriptor(
        '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,
 )