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

Rename to AugmentTransform

Browse Source
This commit is contained in:
henryruhs
2025-03-03 18:29:51 +01:00
parent dd0a2fe649
commit ceb3c0cfdf
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
@@ -38,7 +38,7 @@ class DynamicDataset(Dataset[Tensor]):
[
transforms.ToPILImage(),
transforms.Resize((256, 256), interpolation = transforms.InterpolationMode.BICUBIC),
AlterTransform(),
AugmentTransform(),
transforms.ToTensor(),
WarpTransform(self.warp_template),
transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))
@@ -68,7 +68,7 @@ class DynamicDataset(Dataset[Tensor]):
return source_tensor, target_tensor
class AlterTransform:
class AugmentTransform:
def __init__(self) -> None:
self.transforms = self.compose_transforms()