fix: Pascal Context ignore pixels mapped to class 0 instead of 255

[andrewwhitecdw]

Problem

fix: Pascal Context ignore pixels mapped to class 0 instead of 255

Fix

Replace:

        index = np.digitize(mask.ravel(), _mapping, right=True)
        return _key[index].reshape(mask.shape)

with:

        index = np.digitize(mask.ravel(), _mapping)
        return _key[index].reshape(mask.shape)

Files changed

• datasets/prepare_pascal_ctx_sem_seg.py