diff --git a/embedding_converter/src/data_loader.py b/embedding_converter/src/data_loader.py index 738abbc..55eeed8 100644 --- a/embedding_converter/src/data_loader.py +++ b/embedding_converter/src/data_loader.py @@ -22,7 +22,8 @@ class DataLoaderRecognition(Dataset[torch.Tensor]): def __len__(self) -> int: return len(self.image_paths) - def compose_transforms(self) -> transforms: + @staticmethod + def compose_transforms() -> transforms: return transforms.Compose( [ transforms.ToPILImage(), diff --git a/face_swapper/src/data_loader.py b/face_swapper/src/data_loader.py index bcc3b15..1eaadcc 100644 --- a/face_swapper/src/data_loader.py +++ b/face_swapper/src/data_loader.py @@ -39,7 +39,8 @@ class DataLoader(TensorDataset): image_path_set[directory_path] = image_paths return image_paths, image_path_set - def compose_transforms(self) -> transforms: + @staticmethod + def compose_transforms() -> transforms: return transforms.Compose( [ transforms.ToPILImage(),