import cv2 import os ROOT_PATH='E:\keshe\dataset\project3\project3_data' files=os.listdir(ROOT_PATH) print("需要处理文件的总数量:",len(files)) #任务4:统一图像文件格式 DATA_PATH='E:\keshe\dataset\project3\project3_geshi4' if not os.path.exists(DATA_PATH): os.mkdir(DATA_PATH) cnt=0 for file in files: filename = os.path.join(ROOT_PATH,file) img = cv2.imread (filename,-1) if img is None: continue newfile='{:0>6d}'.format(cnt)+".jpg" cnt+=1 dst_filename=os.path.join(DATA_PATH,newfile) cv2.imwrite(dst_filename,img) print("放入DATA文件夹的文件数量:",cnt)
时间: 2023-07-15 13:11:05 浏览: 160
数据库课程设计_keshe.zip
这段代码的作用是将指定文件夹中的图像文件转化为统一的 jpg 格式,并保存到指定文件夹中。具体实现过程如下:
1. 引入必要的库:cv2 和 os。
2. 定义 ROOT_PATH 和 DATA_PATH 分别为原始图像文件夹和处理后的图像文件夹的路径。
3. 使用 os.listdir() 函数获取原始图像文件夹中所有的文件名。
4. 使用 len() 函数统计总文件数量,并输出。
5. 判断处理后的图像文件夹是否存在,如果不存在则创建。
6. 遍历原始图像文件夹中的所有文件:
- 使用 os.path.join() 函数将文件名和路径合并为完整的文件路径。
- 使用 cv2.imread() 函数读取图像文件,-1 参数表示以原始图像的原始格式读取。
- 如果读取失败,就跳过这个文件。
- 使用 '{:0>6d}'.format(cnt) 格式化文件名,保证文件名有六位数字,不足的前面用 0 填充。
- 将 cnt 加 1,为下一个文件做准备。
- 使用 os.path.join() 函数将处理后的图像文件夹路径和新的文件名合并为完整的文件路径。
- 使用 cv2.imwrite() 函数将处理后的图像保存到指定路径。
7. 输出处理后的图像文件夹中的文件数量。
需要注意的是,Windows 系统中路径分隔符为反斜杠 (\),但是这个字符在 Python 中有特殊含义,需要使用转义字符 \ 来转义。所以在定义路径时需要使用两个反斜杠来表示一个。
阅读全文