【Python】——图像预处理操作

import os
from PIL import Image
import glob


# 1. jpg图像resize成256，256大小
def Resize(file, outdir, width, height):
    imgFile = Image.open(file)
    try:
        newImage = imgFile.resize((width, height), Image.BILINEAR)
        newImage.save(os.path.join(outdir, os.path.basename(file)))
    except Exception as e:
        print(e)

for file in glob.glob("./raw_jpg/*.jpg"):  # 图片所在的目录
    Resize(file, "./1_jpgResize", 256, 256)  # 新图片存放的目录


# 2. jpg彩色图像转为灰度图像
def jpgToGray(input_dir):
    for i in a:
        print(i)
        I = Image.open(input_dir + i)
        L = I.convert(L)
        L.save(out_dir + i)

input_dir = ./1_jpgResize/
out_dir = ./2_jpg2gray/
a = os.listdir(input_dir)
jpgToGray(input_dir)


# 3. jpg图像转为bmp图像
def jpgToBmp(imgFile):
  dst_dir = "./3_jpg2bmp"

  for fileName in os.listdir(imgFile):
    if os.path.splitext(fileName)[1] == .jpg:
      name = os.path.splitext(fileName)[0]
      newFileName = name + ".bmp"

      img = Image.open(imgFile + "/" + fileName)
      img.save(dst_dir+"/"+newFileName)

imgFile = "./2_jpg2gray"
jpgToBmp(imgFile)


# 4. png图像转为bmp图像
def pngToBmp(imgFile):
  dst_dir = "./4_png2bmp"

  for fileName in os.listdir(imgFile):
    if os.path.splitext(fileName)[1] == .png:
      name = os.path.splitext(fileName)[0]
      newFileName = name + ".bmp"

      img = Image.open(imgFile + "/" + fileName)
      img.save(dst_dir+"/"+newFileName)

imgFile = "./raw_png"
pngToBmp(imgFile)