aa7f5bc6fef400e5d7ef411d9af930637f62c547,src/rastervision/data/label_source/semantic_segmentation_raster_source.py,SemanticSegmentationRasterSource,get_labels,#SemanticSegmentationRasterSource#Any#,60

Before Change

        if self.rgb:
            return self.class_transformer.rgb_to_class(img)
        else:
            labels = img[:, :, 0].reshape(img.shape[0], img.shape[1])
            return labels

After Change

        if window is None:
            raw_labels = self.source.get_raw_image_array()
        else:
            raw_labels = self.source.get_raw_chip(window)

        if self.class_transformer is not None:
            labels = self.class_transformer.rgb_to_class(raw_labels)
        else:
            labels = np.squeeze(raw_labels)

        return labels

Instances