fe8fee4f05e08a6448d46b8f70d960f02ef048b8,rastervision_core/rastervision/core/data/label_store/semantic_segmentation_label_store.py,SemanticSegmentationLabelStore,get_labels,#SemanticSegmentationLabelStore#,58

Before Change

        labels = SemanticSegmentationLabels()
        extent = self.source.get_extent()
        raw_labels = self.source.get_raw_chip(extent)
        label_arr = (np.squeeze(raw_labels) if self.class_trans is None else
                     self.class_trans.rgb_to_class(raw_labels))
        labels.set_label_arr(extent, label_arr)
        return labels

After Change

        extent = self.label_raster_source.get_extent()
        raw_labels = self.label_raster_source.get_raw_chip(extent)
        if self.class_transformer is None:
            label_arr = np.squeeze(raw_labels)
        else:
            label_arr = self.class_transformer.rgb_to_class(raw_labels)

        labels = self.empty_labels()
        labels[extent] = label_arr
        return labels

    def get_scores(self) -> SemanticSegmentationLabels:
        Get all scores.

Instances