Skip to content

Commit b87f30a

Browse files
committed
fix cyclegan dataset loader
1 parent efded8e commit b87f30a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensorlayer/files.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -649,7 +649,7 @@ def load_cyclegan_dataset(filename='summer2winter_yosemite', path='data/cyclegan
649649

650650
def load_image_from_folder(path):
651651
path_imgs = load_file_list(path=path, regx='\\.jpg', printable=False)
652-
return tl.vis.read_images(path_imgs, path=path, n_threads=10, printable=False)
652+
return read_images(path_imgs, path=path, n_threads=10, printable=False)
653653
im_train_A = load_image_from_folder(path+"/"+filename+"/trainA")
654654
im_train_B = load_image_from_folder(path+"/"+filename+"/trainB")
655655
im_test_A = load_image_from_folder(path+"/"+filename+"/testA")

0 commit comments

Comments
 (0)