diff options
Diffstat (limited to 'data/image_folder.py')
| -rwxr-xr-x | data/image_folder.py | 3 |
1 files changed, 0 insertions, 3 deletions
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', |