python_opencv ——图片预处里（二）

参考链接：https://blog.csdn.net/qq_35328403/article/details/105352525

1、修改图片格式，并重命名图片名称

import os


class BatchRename():  #定义一个重命名的类
    def __init__(self):
        self.path = 'classification_pic/fatigue driving'

    def rename(self):
        filelist = os.listdir(self.path)
        total_num = len(filelist)
        i = 0
        # for item in filelist:
        #     if item.endswith('.bmp'):   #将bmp格式的图片改为jpg
        #         src = os.path.join(os.path.abspath(self.path), item)
        #         dst = os.path.join(os.path.abspath(self.path), 'Judy' + str(i) + '.jpg') #重新命名并改格式
        #         try:
        #             os.rename(src, dst)
        #             print('converting %s to %s ...' % (src, dst))
        #             i = i + 1
        #         except:
        #             continue

        for item in filelist:
            if item.endswith('.jpg'):
                src = os.path.join(os.path.abspath(self.path), item)
                dst = os.path.join(os.path.abspath(self.path), 'pic_' + str(i) + '.jpg')  ##重新命名
                try:
                    os.rename(src, dst)
                    print('converting %s to %s ...' % (src, dst))
                    i = i + 1
                except:
                    continue

if __name__ == '__main__':
    demo = BatchRename()
    demo.rename()

2、统一图片大小

from PIL import Image
import os.path
import glob

def convertSize(jpgfile, outdir, width=416, height=416):  #图片的大小  416*416
    img=Image.open(jpgfile)
    try:
        new_img = img.resize((width, height), Image.BILINEAR)
        if new_img.mode == 'P':
            new_img = new_img.convert("RGB")
        if new_img.mode == 'RGBA':
            new_img = new_img.convert("RGB")
        new_img.save(os.path.join(outdir, os.path.basename(jpgfile)))
    except Exception as e:
        print(e)

for jpgfile in glob.glob("frame_output/23/*.jpg"):  #修改该文件夹下的jpg图片
    convertSize(jpgfile,"Pic_SameSize/23")  #另存为的文件夹路径