方法一
利用os模块中的walk函数
for root, dir, files in os.walk(dir):
for file in files:
srcImg = cv2.imread(root_path + "data1" + "/" + str(file))
roiImg = srcImg[5:7, 960:962]
cv2.imwrite(root_path + "Image" + "/" + str(file), roiImg)
方法2
利用glob模块来搜索
for idx,path in enumerate(glob.glob("./data/original"+"/*.jpg")):
img=cv2.imread(path,-1)
cv2.imwrite("./data/result/0-%d.jpg"%(idx+1),img)
补充
1.如果需要多个以上格式的文件,不同于方法1,可以使用以下方法
file.endswith('jpg') or file.endswith('png')
2.os模块常用功能,完成对目录/路径的一些操作。
方法3
当需要读取所有文件时(包括普通文件和文件夹)
可以使用os
模块下的listdir
函数
files=os.listdir(path)
然后可以利用os.path.isdir
或者os.path.isfile
来判断