CalvinBackup/facefusion-labs
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion-labs.git synced 2026-05-22 23:59:40 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix same and equal

Browse Source
This commit is contained in:
henryruhs
2025-03-03 12:21:47 +01:00
parent d93daf9e5f
commit 9dc1031fa5
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
face_swapper/src/dataset.py
+2 -2
View File
@@ -55,12 +55,12 @@ class DynamicDataset(Dataset[Tensor]):
target_tensor = self.transforms(target_tensor)
return source_tensor, target_tensor
def prepare_same_batch(self, source_path : str) -> Batch:
def prepare_equal_batch(self, source_path : str) -> Batch:
source_tensor = io.read_image(source_path)
source_tensor = self.transforms(source_tensor)
return source_tensor, source_tensor
def prepare_equal_batch(self, source_path : str) -> Batch:
def prepare_same_batch(self, source_path : str) -> Batch:
target_directory_path = os.path.dirname(source_path)
target_file_name_and_extension = random.choice(os.listdir(target_directory_path))
target_path = os.path.join(target_directory_path, target_file_name_and_extension)