当需要对图片进行格式的修改,例如将图片都改为jpg格式,代码如下
import osimport sysimport cv2import numpy as npdef listfiles(rootDir):list_dirs = os.walk(rootDir) #读取rootDir中的内容for root, dirs, files in list_dirs: # 遍历文件夹下的图片for d in dirs:print((os.path.join(root, d)))for f in files:fileid = f.split('.')[0] # 获得图片的名字,不含后缀filepath = os.path.join(root, f) print(filepath)try:src = cv2.imread(filepath, 1) # 读取原始图片,数据会加载到内存中print("src=", filepath, src.shape)os.remove(filepath) # 移除原来的图片cv2.imwrite(os.path.join(root, fileid + ".jpg"), src) # 保存经过格式转换的图片except:os.remove(filepath)continuepath = "./Emotion_Recognition_File/img_type_test/" # 输入图片路径即可,可以在这个文件夹下放置各种后缀名的图片,代码会将所有图片统一成 jpg 格式listfiles(path)