提交 794f8143 编写于 作者: B Botty Dimanov 提交者: François Chollet

Handle capitalised extensions in list_pictures (#10220)

#10219
上级 84aa7b5f
......@@ -486,7 +486,7 @@ def load_img(path, grayscale=False, target_size=None,
def list_pictures(directory, ext='jpg|jpeg|bmp|png|ppm'):
return [os.path.join(root, f)
for root, _, files in os.walk(directory) for f in files
if re.match(r'([\w]+\.(?:' + ext + '))', f)]
if re.match(r'([\w]+\.(?:' + ext + '))', f.lower())]
class ImageDataGenerator(object):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册