Skip to content
Snippets Groups Projects

First version of dataset class

Merged
s184058requested to merge
data into main
All threads resolved!
1 file
+ 2
2
Compare changes
  • Side-by-side
  • Inline
+ 2
2
@@ -49,9 +49,9 @@ class Dataset(torch.utils.data.Dataset):
@@ -49,9 +49,9 @@ class Dataset(torch.utils.data.Dataset):
self.sample_images = [file for file in sorted((path / "images").iterdir())]
self.sample_images = [file for file in sorted((path / "images").iterdir())]
self.sample_targets = [file for file in sorted((path / "labels").iterdir())]
self.sample_targets = [file for file in sorted((path / "labels").iterdir())]
def __len__(self):
assert len(self.sample_images) == len(self.sample_targets)
assert len(self.sample_images) == len(self.sample_targets)
 
 
def __len__(self):
return len(self.sample_images)
return len(self.sample_images)
def __getitem__(self, idx):
def __getitem__(self, idx):
Loading