diff options
Diffstat (limited to 'data/image_folder.py')
| -rw-r--r-- | data/image_folder.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/data/image_folder.py b/data/image_folder.py index 44e15cb..898200b 100644 --- a/data/image_folder.py +++ b/data/image_folder.py @@ -1,9 +1,9 @@ -################################################################################ +############################################################################### # Code from # https://github.com/pytorch/vision/blob/master/torchvision/datasets/folder.py # 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 @@ -45,7 +45,8 @@ class ImageFolder(data.Dataset): imgs = make_dataset(root) if len(imgs) == 0: raise(RuntimeError("Found 0 images in: " + root + "\n" - "Supported image extensions are: " + ",".join(IMG_EXTENSIONS))) + "Supported image extensions are: " + + ",".join(IMG_EXTENSIONS))) self.root = root self.imgs = imgs |