python read image directory code example

Example 1: python get all images in directory

import glob

for filename in glob.iglob(root_dir + '**/*.jpg', recursive=True):
    # Do something

Example 2: get all type of image in folder python

import cv2
import glob

imdir = 'path/to/files/'
ext = ['png', 'jpg', 'gif']    # Add image formats here

files = []
[files.extend(glob.glob(imdir + '*.' + e)) for e in ext]

images = [cv2.imread(file) for file in files]