From 40dcef7fc78713382a966066e7f5d9eedd5589f2 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Fri, 21 Feb 2025 01:16:46 +0100 Subject: [PATCH] Modernize data loader, remove read image helper --- face_swapper/src/data_loader.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/face_swapper/src/data_loader.py b/face_swapper/src/data_loader.py index 81f73a3..50f3f27 100644 --- a/face_swapper/src/data_loader.py +++ b/face_swapper/src/data_loader.py @@ -19,7 +19,7 @@ class DataLoaderVGG(TensorDataset): self.transforms = self.compose_transforms() def __getitem__(self, index : int) -> Batch: - source_image_path = self.image_paths.get(index) + source_image_path = self.image_paths[index] if random.random() > self.same_person_probability: return self.prepare_same_person(source_image_path)