前言

用PIL.Image 来转换通道

提示：以下是本篇文章正文内容，下面案例可供参考

一、单通道转三通道

#单通道转三通道
def one_to_three():
    img_path = r'./test_imgs'
    save_img_path = r'./test_imgs'
    for file in os.listdir(img_path):
        image = Image.open(os.path.join(img_path, file))
        image=image.convert("RGB")
        image.save(os.path.join(save_img_path, file))

二、三通道转单通道

def three_to_one():
    img_path = r'./test_imgs'
    save_img_path = r'./test_imgs'
    for file in os.listdir(img_path):
        image = Image.open(os.path.join(img_path, file))
        image=image.convert("L")
        image.save(os.path.join(save_img_path, file))

三、检查

def check():
    img_path = r'./test_imgs'
    print("begin:")
    for file in os.listdir(img_path):
        image = Image.open(os.path.join(img_path, file))
        print(np.array(image).shape)

总结

import cv2
from PIL import Image
import os
import numpy as np

def one_to_three():
    img_path = r'./test_imgs'
    save_img_path = r'./test_imgs'
    for file in os.listdir(img_path):
        image = Image.open(os.path.join(img_path, file))
        image=image.convert("RGB")
        image.save(os.path.join(save_img_path, file))

def three_to_one():
    img_path = r'./test_imgs'
    save_img_path = r'./test_imgs'
    for file in os.listdir(img_path):
        image = Image.open(os.path.join(img_path, file))
        image=image.convert("L")
        image.save(os.path.join(save_img_path, file))

def check():
    img_path = r'./test_imgs'
    print("begin:")
    for file in os.listdir(img_path):
        image = Image.open(os.path.join(img_path, file))
        print(np.array(image).shape)

if __name__=="__main__":
    one_to_three()
    check()