NIQE python代码

NIQE（Naturalness Image Quality Evaluator）是一种用于评估图像自然度质量的算法。以下是一个使用Python实现NIQE算法的示例代码：

import numpy as np
import cv2

def compute_niqe(image_path):
    # 读取图像
    image = cv2.imread(image_path, cv2.IMREAD_GRAYSCALE)
    
    # 计算图像的梯度
    gradient_x = cv2.Sobel(image, cv2.CV_64F, 1, 0, ksize=3)
    gradient_y = cv2.Sobel(image, cv2.CV_64F, 0, 1, ksize=3)
    gradient_magnitude = np.sqrt(gradient_x**2 + gradient_y**2)
    
    # 计算局部对比度
    lc = cv2.GaussianBlur(gradient_magnitude, (7, 7), 0)
    
    # 计算图像的自相关
    autocorr = np.correlate(image.flatten(), image.flatten(), 'valid')[0]
    
    # 计算NIQE得分
    niqe_score = np.log10(np.var(gradient_magnitude) * np.var(lc) / autocorr)
    
    return niqe_score

# 示例用法
image_path = 'path/to/image.jpg'
score = compute_niqe(image_path)
print("NIQE Score:", score)

请注意，以上代码仅提供了NIQE算法的基本实现，对于更复杂的图像处理和评估任务，可能需要进一步调整和优化。