correct splits for attacked datasets

Author: maxdreyer

datasets/bone_attacked.py

@@ -78,3 +78,11 @@ def __getitem__(self, idx):
             image = self.augmentation(image)
 
         return image.float(), bone_age
+
+    def get_subset_by_idxs(self, idxs):
+        subset = super().get_subset_by_idxs(idxs)
+        subset.artifact_labels = self.artifact_labels[np.array(idxs)]
+        subset.artifact_ids = np.where(subset.artifact_labels)[0]
+        subset.sample_ids_by_artifact = {"artificial": subset.artifact_ids}
+        subset.clean_sample_ids = [i for i in range(len(subset)) if i not in subset.artifact_ids]
+        return subset

datasets/isic_attacked.py

@@ -78,4 +78,13 @@ def __getitem__(self, i):
             img = self.augmentation(img)
         columns = self.metadata.columns.to_list()
         target = torch.Tensor([columns.index(row[row == 1.0].index[0]) - 1 if self.train else 0]).long()[0]
-        return img, target
+        return img, target
+
+    def get_subset_by_idxs(self, idxs):
+        subset = super().get_subset_by_idxs(idxs)
+        subset.artifact_labels = self.artifact_labels[np.array(idxs)]
+
+        subset.artifact_ids = np.where(subset.artifact_labels)[0]
+        subset.sample_ids_by_artifact = {"artificial": subset.artifact_ids}
+        subset.clean_sample_ids = [i for i in range(len(subset)) if i not in subset.artifact_ids]
+        return subset