diff options
| author | Volpeon <git@volpeon.ink> | 2022-09-27 22:16:13 +0200 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2022-09-27 22:16:13 +0200 |
| commit | a8a5abae42f6f42056cc27e0cf5313aab080c3a7 (patch) | |
| tree | 32c163bbc58aa2f827c5ba5108df81dc14fbe130 /data | |
| parent | Incorporate upstream changes (diff) | |
| download | textual-inversion-diff-a8a5abae42f6f42056cc27e0cf5313aab080c3a7.tar.gz textual-inversion-diff-a8a5abae42f6f42056cc27e0cf5313aab080c3a7.tar.bz2 textual-inversion-diff-a8a5abae42f6f42056cc27e0cf5313aab080c3a7.zip | |
Various improvements, added inference script
Diffstat (limited to 'data')
| -rw-r--r-- | data/dreambooth/csv.py | 1 |
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): | |||
| 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 | ] |