Python计算机视觉读书笔记_04：获得文件夹中所有图片文件名，并生成列表

在图像处理中，我们经常会遇到要读取文件夹中所有图片的操作，这时，我们想要获得的是带有路径的图片文件名，接下来就记录一下如何得到存放所有图片文件名的列表。

首先，我们的图片路径如下所示，我们要获得 ".jpg" 结尾的所有图片名。

然后建立一个imtools.py文件，代码如下：

import os


def get_imlist(path):
    """返回目录中所有jpg图像的文件名列表"""
    return [os.path.join(path, f) for f in os.listdir(path) if f.endswith('.jpg')]

然后新建main.py文件：

import imtools

list = imtools.get_imlist('D:\\AnacondaProject\\OpenCV')
for i in list:
    print(i, '\n')
print(len(list))

运行结果如下所示：