对比图像内容差异并标记

2023-08-12 | 阅读

文章目录

1. 让ChatGPT为我们工作
2. 验证程序

今天在一个群里，一个兄弟发起了一个讨论，两个线路板的图片，怎么能对比出两者的不同之处。

这里我暂时假设两张图片都是像素大小差不多标准化的图像，内容只有细微差异（想一下如果差异太大，那还对比啥呢？）

大胆猜测一下大概率用于生产过程中的质量控制或者分类等场景；只要找到些许不同，人眼应该就能一眼判断是否需要继续下一步操作，所以找到不同之处之后还需要标记。

这不就是以前我们常玩的游戏找不同吗？

让ChatGPT为我们工作

因为没有专门学习这方面的知识，所以尝试问了一下 ChatGPT ，并检查它生成的代码是否能直接运行。

需求：
输入两张图，对比差异，标记差异

由于我的确不知道需要使用什么库去做就不限定使用库了，直接说需求。

问：
请使用 Python 编写一个程序，对比两张图片，并在后一张图上标记出差异

ChatGPT答:
要使用Python比较两张图并在后一张图上标记差异，你可以使用OpenCV和Matplotlib库。下面是一个简单的示例代码：

import cv2
import numpy as np
import matplotlib.pyplot as plt

def compare_images(image1, image2):
    # 加载两张图像
    img1 = cv2.imread(image1)
    img2 = cv2.imread(image2)

    # 将图像转换为灰度图
    gray1 = cv2.cvtColor(img1, cv2.COLOR_BGR2GRAY)
    gray2 = cv2.cvtColor(img2, cv2.COLOR_BGR2GRAY)

    # 计算两张图像的差异
    diff = cv2.absdiff(gray1, gray2)
    _, threshold = cv2.threshold(diff, 30, 255, cv2.THRESH_BINARY)

    # 查找差异的轮廓
    contours, _ = cv2.findContours(threshold, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)

    # 在第二张图像上绘制差异的轮廓
    marked_image = img2.copy()
    for contour in contours:
        (x, y, w, h) = cv2.boundingRect(contour)
        cv2.rectangle(marked_image, (x, y), (x + w, y + h), (0, 255, 0), 2)

    # 显示图像
    plt.subplot(1, 3, 1)
    plt.imshow(cv2.cvtColor(img1, cv2.COLOR_BGR2RGB))
    plt.title('Image 1')
    plt.axis('off')

    plt.subplot(1, 3, 2)
    plt.imshow(cv2.cvtColor(img2, cv2.COLOR_BGR2RGB))
    plt.title('Image 2')
    plt.axis('off')

    plt.subplot(1, 3, 3)
    plt.imshow(cv2.cvtColor(marked_image, cv2.COLOR_BGR2RGB))
    plt.title('Differences')
    plt.axis('off')

    plt.tight_layout()
    plt.show()

# 比较两张图像并标记出差异
compare_images('image1.png', 'image2.png')