def __init__()

in scripts/dataset/dataset.py [0:0]

• 14 lines of code
• 3 McCabe index (conditional complexity)

    def __init__(self, in_dir, transform):
        super(SegmentationDataset, self).__init__()

        self.in_dir = in_dir
        self.transform = transform

        # Find all images
        self.images = []
        for img_path in chain(
            *(
                glob.iglob(path.join(self.in_dir, ext))
                for ext in SegmentationDataset._EXTENSIONS
            )
        ):
            _, name_with_ext = path.split(img_path)
            idx, _ = path.splitext(name_with_ext)
            self.images.append({"idx": idx, "path": img_path})