使用给定文件夹中的 imread 加载所有图像

Question

在 OpenCV 中加载和保存图像非常有限，所以…从给定文件夹加载所有图像的首选方法是什么？我是否应该在该文件夹中搜索扩展名为 .png 或 .jpg 的文件，存储名称并对每个文件使用 imread ？或者，还有更好的方法？

原文由 user107986 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

为什么不尝试加载文件夹中的所有文件？如果OpenCV打不开，唉。继续下一步。 cv2.imread() 返回 None 如果图像无法打开。有点奇怪，它没有引发异常。

 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

原文由 derricw 发布，翻译遵循 CC BY-SA 3.0 许可协议