Loading all images using imread from a given folder

Question

Loading and saving images in OpenCV is quite limited, so... what is the preferred ways to load all images from a given folder? Should I search for files in that folder with .png or .jpg extensions, store the names and use imread with every file? Or is there a better way?

Answer 1

Why not just try loading all the files in the folder? If OpenCV can't open it, oh well. Move on to the next. cv2.imread() returns None if the image can't be opened. Kind of weird that it doesn't raise an exception.

import cv2 import os  def load_images_from_folder(folder):     images = []     for filename in os.listdir(folder):         img = cv2.imread(os.path.join(folder,filename))         if img is not None:             images.append(img)     return images