最近在做一个和图像有关的项目,需要遍历文件夹中的图片,这里可以使用两种方法:
1)使用python的glob模块:
import cv2
import glob as gb
img_path = gb.glob(r'D:\pictures\*.jpg') #此处为window系统,*.jpg代表全部jpg文件
count = 0
for i in img_path: #i为全部jpg文件集合中的一个jpg文件
count += 1
img = cv2.imread(i)
print(type(i))
print(i)
if count == 3:
break
结果为:
2)使用os模块下的os.listdir():
import os
img_path = os.listdir(r'D:\pictures') #此处为Windows系统,括号中为文件夹的路径
c = 0
for i in img_path:
c = c+1
img = cv2.imread(r'D:\pictures\%s'%(i))
print(type(i)) #i为文件夹下带后缀的文件名
print(i)
if c == 3:
break