方法一
利用os模塊中的walk函數(shù)
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
函數(shù)
files=os.listdir(path)
然后可以利用os.path.isdir
或者os.path.isfile
來判斷