From e6858e35f0a08c6139c133122d222d0d85e8005d Mon Sep 17 00:00:00 2001 From: junyanz Date: Mon, 12 Jun 2017 23:52:56 -0700 Subject: update dataset mode --- data/image_folder.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) (limited to 'data/image_folder.py') 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 -- cgit v1.2.3-70-g09d2