From da51c5336d5ff57ea7b953c171bdda0e3f07be32 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Fri, 21 Feb 2025 08:39:08 +0100 Subject: [PATCH] This can be marked static --- embedding_converter/src/data_loader.py | 3 ++- face_swapper/src/data_loader.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) 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(),