python中利用opencv简单做图片比对

最新推荐文章于 2025-10-25 11:06:07 发布

原创最新推荐文章于 2025-10-25 11:06:07 发布 · 6.1k 阅读

5 ·

CC 4.0 BY-SA版权

文章标签：

#python #图片 #opencv

machine learning 专栏收录该内容

3 篇文章

订阅专栏

这篇博客介绍如何在Python环境下使用OpenCV库对二值单通道图片进行比对。作者提供了两种方法：一是通过计算图片矩阵的特征值并设定阈值判断；二是计算图片中1的总数（sum和）进行比较。

python环境中，利用opencv对二值单通道图片进行比对

下面代码中利用了两种比对的方法，一对图片矩阵（m x m）求解特征值，通过比较特征值是否在一定的范围内，判断图片是否相同。二对图片矩阵（m x m）中1求和，通过比较sum和来比较图片。

# -*- coding: utf-8 -*-
import cv2  as  cv
import numpy as  np
import os
file_dir_a='C:\Users\wt\Desktop\data\image1\\'
file_dir_b='C:\Users\wt\Desktop\data\image\\'
savepath='.\'

all_file_name_a=os.listdir(file_dir_a)
all_file_name_b=os.listdir(file_dir_b)
image_all_a=[]
image_all_b=[]
for name in all_file_name_a:
    image_one=[]
    image = cv.imread(file_dir_a+name, cv.IMREAD_GRAYSCALE)
    """arg是计算输入图片矩阵的特征值，通过对特征值的比较来实现图片的比对
    """
    # arg=np.linalg.eigvals(image) 
    """arg是计算输入二值图片矩阵中1的个数，通过1的总数来实现图片的比对
    """
    arg=sum(image)
    image_one.append(name)
    image_one.append(arg)
    image_all_a.append(image_one)#将一个图片的信息写入
    print '读入a'
# np.save('img_a.npy',image_all_a)
for name in all_file_name_b:
    image_one=[]
    image = cv.imread(file_dir_b+name, cv.IMREAD_GRAYSCALE)
    """同上
    """
    # arg=np.linalg.eigvals(image)
    arg=sum(image)
    image_one.append(name)
    image_one.append(arg)
    image_all_b.append(image_one)#将一个图片的信息写入
    print '读入b'
# np.save('img_b.npy',image_all_b)
print '开始比较'
result_all=[]
for a in image_all_a:  #比较小的
    result = []
    for b in image_all_b:
        # print sum(a[1]-b[1])
        if abs(sum(a[1]-b[1]))<0.00001:
            result.append(a[0])
            result.append(b[0])
            result_all.append(result)
print '比较结束'
print result_all

np.save('match_result1.npy',result_all)