CIEDE2000的Python实现批量比较两个文件夹中图片的色差

W&W_O.o

于 2024-03-26 12:54:02 发布

阅读量409

点赞数 2

文章标签： python 开发语言

本文链接：https://blog.csdn.net/qq_45359549/article/details/137042769

版权

from skimage.color import deltaE_ciede2000
import cv2
import os
import numpy as np

folder1 = "folder1"   #这里修改为你的文件夹路径
folder2 = "folder2"   #这里修改为你的文件夹路径

files1 = os.listdir(folder1)
files2 = os.listdir(folder2)

ciede_values_list = []

for idx, (file1, file2) in enumerate(zip(files1, files2), start=1):
    im1 = cv2.imread(os.path.join(folder1, file1))
    im2 = cv2.imread(os.path.join(folder2, file2))

    im1_lab = cv2.cvtColor(im1, cv2.COLOR_BGR2LAB).astype(np.float32)
    im2_lab = cv2.cvtColor(im2, cv2.COLOR_BGR2LAB).astype(np.float32)
    ciede_values = deltaE_ciede2000(im1_lab, im2_lab).mean()

    ciede_values_list.append(ciede_values)
    print(f"Processing pair {idx}: {file1} and {file2}")

avg_ciede = np.mean(ciede_values_list)
print(f"Average CIEDE_2000: {avg_ciede}")

使用前先在虚拟环境中导入必要的包，为了使用方便，可以将该py代码和两个文件夹放在同一目录内。