Skip to content
This repository was archived by the owner on Jul 2, 2021. It is now read-only.

Add ADE20K dataset #429

Merged
merged 17 commits into from
Oct 6, 2017
Prev Previous commit
Next Next commit
Fix test
  • Loading branch information
mitmul committed Sep 28, 2017
commit 9f701025e64b759ffef02fe717911659a9acbd50
4 changes: 2 additions & 2 deletions tests/datasets_tests/ade20k_tests/test_ade20k.py
Original file line number Diff line number Diff line change
@@ -30,8 +30,8 @@ def test_ade20k_dataset(self):
assert_is_semantic_segmentation_dataset(
self.dataset, len(ade20k_label_names), n_example=10)
else:
idx = np.random.permutation(np.arange(len(self.dataset)))
for i in idx[:10]:
indices = np.random.permutation(np.arange(len(self.dataset)))
for i in indices[:10]:
img = self.dataset[i]
assert_is_image(img, color=True)