diff options
Diffstat (limited to 'data')
| -rwxr-xr-x | data/aligned_dataset.py | 4 | ||||
| -rwxr-xr-x | data/image_folder.py | 3 |
2 files changed, 0 insertions, 7 deletions
diff --git a/data/aligned_dataset.py b/data/aligned_dataset.py index 50390f3..a0c9a0a 100755 --- a/data/aligned_dataset.py +++ b/data/aligned_dataset.py @@ -1,13 +1,9 @@ ### Copyright (C) 2017 NVIDIA Corporation. All rights reserved. ### Licensed under the CC BY-NC-SA 4.0 license (https://creativecommons.org/licenses/by-nc-sa/4.0/legalcode). import os.path -import random -import torchvision.transforms as transforms -import torch from data.base_dataset import BaseDataset, get_params, get_transform, normalize from data.image_folder import make_dataset from PIL import Image -import numpy as np class AlignedDataset(BaseDataset): def initialize(self, opt): diff --git a/data/image_folder.py b/data/image_folder.py index 16a447c..df0141f 100755 --- a/data/image_folder.py +++ b/data/image_folder.py @@ -4,12 +4,9 @@ # Modified the original code so that it also loads images from the current # directory as well as the subdirectories ############################################################################### - import torch.utils.data as data - from PIL import Image import os -import os.path IMG_EXTENSIONS = [ '.jpg', '.JPG', '.jpeg', '.JPEG', |