Add metadata dict

microsoft · Feb 19, 2025 · d727e47 · d727e47
1 parent 4c38d76
commit d727e47
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/torchgeo/datasets/ssl4eo.py b/torchgeo/datasets/ssl4eo.py
@@ -499,11 +499,13 @@ def __getitem__(self, index: int) -> dict[str, Tensor]:
 
         sample = {
             'image': torch.cat(images),
-            'x': torch.tensor(xs),
-            'y': torch.tensor(ys),
-            't': torch.tensor(ts),
-            'wavelength': torch.tensor(wavelengths),
-            'res': torch.tensor(10),
+            'metadata': {
+                'x': torch.tensor(xs),
+                'y': torch.tensor(ys),
+                't': torch.tensor(ts),
+                'wavelength': torch.tensor(wavelengths),
+                'res': torch.tensor(10),
+            },
         }
 
         if self.transforms is not None: