140b558954c61e77551ff93545864b78b086e1b1,chainercv/utils/image/read_label.py,,read_label,#Any#Any#,5

Before Change

        return img[np.newaxis]
    else:
        // transpose (H, W, C) -> (C, H, W)
        return img.transpose((2, 0, 1))

After Change

    if img.ndim == 2:
        return img
    elif img.shape[2] == 1:
        return img[:, :, 0]
    else:
        raise ValueError("Color image can"t be accepted as label image.")

Instances