Fix duplicate transform bug (#268)

Clay-foundation · Jun 7, 2024 · 7f18748 · 7f18748
1 parent 49fe176
commit 7f18748
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/finetune/segment/chesapeake_datamodule.py b/finetune/segment/chesapeake_datamodule.py
@@ -89,10 +89,6 @@ def __getitem__(self, idx):
         label_mapping = {1: 0, 2: 1, 3: 2, 4: 3, 5: 4, 6: 5, 15: 6}
         remapped_label = np.vectorize(label_mapping.get)(label)
 
-        # Apply transformations
-        if self.transform:
-            chip = self.transform(torch.from_numpy(chip))
-
         sample = {
             "pixels": self.transform(torch.from_numpy(chip)),
             "label": torch.from_numpy(remapped_label[0]),