1 files changed, 1 insertions, 0 deletions
diff --git a/data/dreambooth/csv.py b/data/dreambooth/csv.py
index e70c068..14c13bb 100644
--- a/data/dreambooth/csv.py
+++ b/data/dreambooth/csv.py
@@ -117,6 +117,7 @@ class CSVDataset(Dataset):
            [
                transforms.Resize(size, interpolation=transforms.InterpolationMode.BILINEAR),
                transforms.CenterCrop(size) if center_crop else transforms.RandomCrop(size),
+                transforms.RandomHorizontalFlip(),
                transforms.ToTensor(),
                transforms.Normalize([0.5], [0.5]),
            ]

diff --git a/data/dreambooth/csv.py b/data/dreambooth/csv.py index e70c068..14c13bb 100644 --- a/data/dreambooth/csv.py +++ b/data/dreambooth/csv.py
@@ -117,6 +117,7 @@ class CSVDataset(Dataset):
117	[	117	[
118	transforms.Resize(size, interpolation=transforms.InterpolationMode.BILINEAR),	118	transforms.Resize(size, interpolation=transforms.InterpolationMode.BILINEAR),
119	transforms.CenterCrop(size) if center_crop else transforms.RandomCrop(size),	119	transforms.CenterCrop(size) if center_crop else transforms.RandomCrop(size),
		120	transforms.RandomHorizontalFlip(),
120	transforms.ToTensor(),	121	transforms.ToTensor(),
121	transforms.Normalize([0.5], [0.5]),	122	transforms.Normalize([0.5], [0.5]),
122	]	123	]