pkmital / CADL

ARCHIVED: Contains historical course materials/Homework materials for the FREE MOOC course on "Creative Applications of Deep Learning w/ Tensorflow" #CADL
https://www.kadenze.com/courses/creative-applications-of-deep-learning-with-tensorflow/info
Apache License 2.0
1.48k stars 732 forks source link

Update utils.py #54

Open pierrevz opened 7 years ago

pierrevz commented 7 years ago

This change limits get_celeb_files to return a maximum of 100 files when the directory 'img_align_celeba' already exists with more than 100 files. The full data-set is over 200,000 files. This code still relies on the files being sequential and in the format '000%03d' which is the original format. Where files are missing, the number of returned files will less than 100.

Note: a future recommendation is to pass a file_quantity parameter to get_celeb_imgs to allow for data-sets of any size.