ee63af01e28b1ac2b31eda1a28aa72920bc9dd6a,deepplantphenomics/semantic_segmentation_model.py,SemanticSegmentationModel,load_dataset_from_directory_with_segmentation_masks,#SemanticSegmentationModel#Any#Any#,319

Before Change

        :param seg_dirname: the path of the directory containing ground-truth binary segmentation masks
        

        image_files = [os.path.join(dirname, name) for name in os.listdir(dirname) if
                       os.path.isfile(os.path.join(dirname, name)) & name.endswith(".png")]

        seg_files = [os.path.join(seg_dirname, name) for name in os.listdir(seg_dirname) if
                     os.path.isfile(os.path.join(seg_dirname, name)) & name.endswith(".png")]

After Change

        self._raw_image_files = loaders.get_dir_images(dirname)
        self._raw_labels = loaders.get_dir_images(seg_dirname)
        if self._with_patching:
            self._raw_image_files, self._raw_labels = self.__autopatch_segmentation_dataset()

        self._total_raw_samples = len(self._raw_image_files)
        self._log("Total raw examples is %d" % self._total_raw_samples)

Instances