1. 程式人生 > >python中利用opencv簡單做圖片比對

python中利用opencv簡單做圖片比對

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)