INNER CODE UNIT · Python
get_image_list
Mukosame/Anime2Sketch · data.py:22
def get_image_list(path):
"""read the paths of valid images from the given directory path
Parameters:
path (str) -- input directory path
"""
assert os.path.isdir(path), '{:s} is not a valid directory'.format(path)
images = []
for dirpath, _, fnames in sorted(os.walk(path)):
for fname in sorted(fnames):
if is_image_file(fname):
img_path = os.path.join(dirpath, fname)
images.append(img_path)
assert images, '{:s} has no valid image file'.format(path)
return images
def get_transform(load_size=0, grayscale=False, method=bic, convert=True):
transform_list = []
if grayscale: