7ab028d2156122a80e7e638e56eafedb9e0d2425,inferno/io/box/bsd500.py,BSD500,__init__,#BSD500#Any#Any#Any#Any#Any#Any#,119

Before Change

                        out.create_dataset("{}/image_data/{}".format(ds, i),
                                           data=img)
                        all_segmentations = sio.loadmat(f)["groundTruth"][0]
                        for subject in range(all_segmentations.shape[0]):
                            label_img = all_segmentations[subject][0][0][0]
                            out.create_dataset("{}/label_data/{}_{}".format(ds, i, subject),
                                               data=label_img)

        self.subject = subject
        self.root_folder = root_folder
        self.split = split

After Change

                        out.create_dataset("{}/image_data/{}".format(ds, i),
                                           data=img)
                        all_segmentations = sio.loadmat(f)["groundTruth"][0]
                        best_gt_index = np.argmax([len(np.unique(s)) for s in all_segmentations])
                        print(all_segmentations)
                        label_img = all_segmentations[:, 0][0][0]
                        out.create_dataset("{}/label_data/{}".format(ds, i),
                                           data=label_img)

Instances