遍历文件夹并保存

代码

导入os包

import os

遍历文件夹

if __name__ == '__main__':
	data_path = "" #需要遍历的文件夹名
	img_list = os.listdir(data_path)
	if len(img_list) == 0:
		print('Data directory is empty.')
		exit()
    for img_name in img_list:
        img = cv2.imread(os.path.join(data_path, img_name)) #之后可以对文件夹中的图片进行形态学操作

将处理后的图片保存到指定文件夹

cv2.imwrite('' + "\\" + img_name, result) #放入for循环中

后话

这里只是简单记录一个实用工具，相信之后肯定会用到很多遍历文件夹的项目。