Reinhard颜色迁移算法python

学不会Jvav

已于 2024-08-02 16:46:36 修改

阅读量116

点赞数 3

文章标签：算法 python opencv

于 2024-08-02 16:46:02 首次发布

本文链接：https://blog.csdn.net/qq_44304221/article/details/140874578

版权

Reinhard颜色迁移算法

应该是处理速度最快的写法了(cpu)

import numpy as np
import cv2


def get_avgstd_cv2(image1, image2):
    avg1, std1 = cv2.meanStdDev(image1)
    avg2, std2 = cv2.meanStdDev(image2)
    return avg1.flatten().astype(np.float32), std1.flatten().astype(np.float32), avg2.flatten().astype(
        np.float32), std2.flatten().astype(np.float32)


def rct(src, dst):
    image = cv2.cvtColor(src, cv2.COLOR_BGR2LAB).astype(np.float32)
    original = cv2.cvtColor(dst, cv2.COLOR_BGR2LAB)

    # 获取均值和标准差
    image_avg, image_std, original_avg, original_std = get_avgstd_cv2(image, original)

    np.subtract(image, image_avg, out=image)
    np.divide(original_std, image_std, out=original_std)
    np.multiply(image, original_std, out=image)
    np.add(image, original_avg, out=image)
    np.clip(image, 0, 255, out=image)

    # 将调整后的图像转换回 BGR 色彩空间
    image = cv2.cvtColor(image.astype(np.uint8), cv2.COLOR_LAB2BGR)

    return image


if __name__ == '__main__':
    p1 = r'test1.jpg'
    p2 = r'test2.jpg'
    p1 = cv2.imread(p1)
    p2 = cv2.imread(p2)
    image = rct(p1, p2)
    cv2.imwrite(r'out.jpg', image)