批量读取并保存图片
针对文件夹里无规则命名的图片
import os
import cv2
def getFileList(dir, Filelist, ext=None):
"""
获取文件夹及其子文件夹中文件列表
输入 dir:文件夹根目录
输入 ext: 扩展名
返回: 文件路径列表
"""
newDir = dir
if os.path.isfile(dir):
if ext is None:
Filelist.append(dir)
else:
if ext in dir[-3:]:
Filelist.append(dir)
elif os.path.isdir(dir):
for s in os.listdir(dir):
newDir = os.path.join(dir, s)
getFileList(newDir, Filelist, ext)
return Filelist
org_img_folder = 'left_1' #选择图片所在文件夹
# 检索文件
imglist = getFileList(org_img_folder, [], 'jpg')
print('本次执行检索到 ' + str(len(imglist)) + ' 张图像\n')
i=1
for imgpath in imglist:
# 对每幅图像执行相关操作,如
img = cv2.resize(imgpath,(600,600))
#按照1,2,3....命名处理后的图片并保存在left_1_undistort文件夹下
cv2.imwrite('left_1_undistort/'+str(i)+'.jpg', img)
i+=1