python实现图像从RGB到Lab空间的转换

本文主要参考来源为：

https://www.cnblogs.com/hrlnw/p/4126017.html

原文是用c++写的，个人仿照其原理进行了python实现

但不足之处在于，不知道转化之后的矩阵能不能保存为图片，查了一些资料说，这种转化结果只是数据形式，不能保存

为图片格式。很怀疑到底能不能这样做，还得继续进行探索。

#coding=utf-8
from PIL import Image
import numpy as np
import os
import math

image_path = r'**\image'

#将图像转换为想要的尺寸
rows, cols, channels = 224, 224, 1

def gamma(im_channel):
    return ((im_channel+0.055)/1.055)**2.4 if im_channel > 0.04045 else im_channel / 12.92

def f(im_channel):
    return im_channel**1/3 if im_channel > 0.008856 else 7.787 * im_channel + 0.137931

def convert_image(image_path, rows,cols, channels,  mode):
    #读入图像
    imgs = os.listdir(image_path)
    num = len(imgs)
    #创建保存结果的矩阵
    L_ = np.zeros((num, rows, cols, channels), dtype=np.float32)
    a_ = np.zeros((num, rows, cols, channels), dtype=np.float32)
    b_ = np.zeros((num, rows, cols, channels), dtype=np.float32)
    
    for i in range(num):
        img = Image.open(im)
        img = img.resize((rows, cols))
        #获取像素
        pix = img.load()
        for x in range(rows):
            for y in range(cols):
                # 获得三个通道值
                r, g, b = pix[x, y]

                #伽马映射
                R = gamma(r/255.0)
                G = gamma(g/255.0)
                B = gamma(b/255.0)
                #转换到XYZ空间
                X = 0.412453 * R + 0.357580 * G + 0.180423 * B
                Y = 0.212671 * R + 0.715160 * G + 0.072169 * B
                Z = 0.019334 * R + 0.119193 * G + 0.950227 * B
                #对XYZ空间进行归一化
                X /= 0.95047
                Y /= 1.0
                Z /= 1.08883

                #对XYZ空间进行映射
                FX = f(X)
                FY = f(Y)
                FZ = f(Z)

                #转换到Lab空间
                L_[x, y] = 116 * FY - 16 if Y > 0.008856 else 903.3 * Y
                a_[x, y] = 500 * (FX - FY)
                b_[x, y] = 200 * (FY - FZ)