Python判断两张图片的相似度

在Python中,判断两张以numpyndarray格式存储的图片的相似度,通常可以通过多种方法来实现,包括但不限于直方图比较、像素差比较、结构相似性指数(SSIM)、特征匹配等。以下是一些常见方法的简要介绍和示例代码。

1. 像素差比较

最直接的方法是计算两张图片对应像素之间的差异。然而,这种方法对图片的旋转、缩放等变换非常敏感。

python 复制代码
import numpy as np

def pixel_difference(img1, img2):
    if img1.shape != img2.shape:
        raise ValueError("Images must have the same shape")
    diff = np.sum(np.abs(img1.astype(np.float32) - img2.astype(np.float32)))
    return diff

# 假设 img1 和 img2 是两个相同尺寸的 ndarray 图片
# result = pixel_difference(img1, img2)
# print(f"Pixel Difference: {result}")

2. 直方图比较

通过比较两张图片的直方图来判断它们的相似度。这种方法对颜色分布敏感,但对空间信息不敏感。

python 复制代码
from skimage.exposure import histogram

def histogram_comparison(img1, img2):
    hist1, bin_idx1 = histogram(img1)
    hist2, bin_idx2 = histogram(img2)
    # 这里可以使用多种方法来比较直方图,如计算欧氏距离、巴氏距离等
    diff = np.linalg.norm(hist1 - hist2)
    return diff

# 假设 img1 和 img2 是两个 ndarray 图片
# result = histogram_comparison(img1, img2)
# print(f"Histogram Difference: {result}")

3. 结构相似性指数(SSIM)

SSIM是一种衡量两幅图片视觉相似度的指标,它考虑了亮度、对比度和结构信息。

python 复制代码
from skimage.metrics import structural_similarity as ssim

def compare_ssim(img1, img2):
    # 确保img1和img2是灰度图,如果不是,需要先转换
    if img1.ndim == 3:
        img1 = img1.mean(axis=2)
    if img2.ndim == 3:
        img2 = img2.mean(axis=2)
    (score, diff) = ssim(img1, img2, full=True)
    return score, diff

# 假设 img1 和 img2 是两个 ndarray 图片
# score, diff = compare_ssim(img1, img2)
# print(f"SSIM: {score}, Diff: {diff}")

注意

  • 上述代码中的img1img2应该是相同尺寸和类型的numpy.ndarray
  • 对于颜色图片,一些方法(如SSIM)可能需要先将图片转换为灰度图或使用其他方式处理颜色通道。
  • 在实际应用中,可能需要根据图片的具体情况和需求选择合适的方法。
相关推荐
erxij8 分钟前
【游戏引擎之路】登神长阶(十四)——OpenGL教程:士别三日,当刮目相看
c++·经验分享·游戏·3d·游戏引擎
Lizhihao_21 分钟前
JAVA-队列
java·开发语言
denghai邓海21 分钟前
红黑树删除之向上调整
python·b+树
HC1825808583225 分钟前
“倒时差”用英语怎么说?生活英语口语学习柯桥外语培训
学习·生活
学习路上_write29 分钟前
FPGA/Verilog,Quartus环境下if-else语句和case语句RT视图对比/学习记录
单片机·嵌入式硬件·qt·学习·fpga开发·github·硬件工程
非概念35 分钟前
stm32学习笔记----51单片机和stm32单片机的区别
笔记·stm32·单片机·学习·51单片机
远望清一色39 分钟前
基于MATLAB边缘检测博文
开发语言·算法·matlab
努力的小雨1 小时前
快速上手 KSQL:轻松与数据库交互的利器
数据库·经验分享
封步宇AIGC1 小时前
量化交易系统开发-实时行情自动化交易-3.4.1.2.A股交易数据
人工智能·python·机器学习·数据挖掘
何曾参静谧1 小时前
「Py」Python基础篇 之 Python都可以做哪些自动化?
开发语言·python·自动化