python_opencv ——图片预处里（二）
参考链接：https://blog.csdn.net/qq_35328403/article/details/105352525
1、修改图片格式，并重命名图片名称
1 import os 2 3 4 class BatchRename(): #定义一个重命名的类 5 def __init__(self): 6 self.path = 'classification_pic/fatigue driving' 7 8 def rename(self): 9 filelist = os.listdir(self.path) 10 total_num = len(filelist) 11 i = 0 12 # for item in filelist: 13 # if item.endswith('.bmp'): #将bmp格式的图片改为jpg 14 # src = os.path.join(os.path.abspath(self.path), item) 15 # dst = os.path.join(os.path.abspath(self.path), 'Judy' + str(i) + '.jpg') #重新命名并改格式 16 # try: 17 # os.rename(src, dst) 18 # print('converting %s to %s ...' % (src, dst)) 19 # i = i + 1 20 # except: 21 # continue 22 23 for item in filelist: 24 if item.endswith('.jpg'): 25 src = os.path.join(os.path.abspath(self.path), item) 26 dst = os.path.join(os.path.abspath(self.path), 'pic_' + str(i) + '.jpg') ##重新命名 27 try: 28 os.rename(src, dst) 29 print('converting %s to %s ...' % (src, dst)) 30 i = i + 1 31 except: 32 continue 33 34 if __name__ == '__main__': 35 demo = BatchRename() 36 demo.rename()
2、统一图片大小
1 from PIL import Image 2 import os.path 3 import glob 4 5 def convertSize(jpgfile, outdir, width=416, height=416): #图片的大小 416*416 6 img=Image.open(jpgfile) 7 try: 8 new_img = img.resize((width, height), Image.BILINEAR) 9 if new_img.mode == 'P': 10 new_img = new_img.convert("RGB") 11 if new_img.mode == 'RGBA': 12 new_img = new_img.convert("RGB") 13 new_img.save(os.path.join(outdir, os.path.basename(jpgfile))) 14 except Exception as e: 15 print(e) 16 17 for jpgfile in glob.glob("frame_output/23/*.jpg"): #修改该文件夹下的jpg图片 18 convertSize(jpgfile,"Pic_SameSize/23") #另存为的文件夹路径
