Skip to content

GitLab

Unverified Commit e225b20a authored by Yukun Zhu's avatar Yukun Zhu Committed by GitHub
Browse files

Update segmentation_dataset.py 

Update ADE20K dataset info.
parent 76d494bd
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
research/deeplab/datasets/segmentation_dataset.py
View file @ e225b20a
...@@ -103,8 +103,8 @@ _ADE20K_INFORMATION = DatasetDescriptor( ...@@ -103,8 +103,8 @@ _ADE20K_INFORMATION = DatasetDescriptor(
'train': 20210, # num of samples in images/training 'train': 20210, # num of samples in images/training
'val': 2000, # num of samples in images/validation 'val': 2000, # num of samples in images/validation
}, },
num_classes=150, num_classes=151,
ignore_label=255, ignore_label=0,
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment