Python 列出文件夹下某类文件后缀
读某个文件夹下所有图片
利用
os.listdir()
和string.endswith()
函数实现。
在进行视觉相关任务编程时, 常常需要读出文件夹下的图片。但有的时候会包含其他后缀的文件,这时候需要将特定后缀的文件名依次读出。
python的字符串提供了一个匹配结尾的函数string.endswith()
,其用法如下:
string.endswith(value, start, end)
参数 | 含义 |
---|---|
value | 字符串 |
start | 检测的起始位置 |
end | 检测的终止位置 |
返回值 | True/False |
如果需要读入的图片为xxx.jpg
那么就可以只用上述函数来判断,下面是一个具体的使用例子:
#从文件夹中读取所有的图片
import os
import cv2
path = 'where/your/photo/stored'
names = os.listdir(path) #这将返回一个所有文件名的列表
for name in names:
if name.endswith('.jpg'): #如果以图片jpg结尾则为真,同样对于png,bmp,txt也适用
img = cv2.imread(path+name)
# Your process code
然后就可以愉快地读入了~~~
icon from easyicon.net