相位相关的模板匹配python代码

以下是相位相关的模板匹配的 Python 代码示例：

import cv2
import numpy as np

# 读取原始图像和模板图像
img = cv2.imread('original_image.jpg', )
template = cv2.imread('template_image.jpg', )

# 计算原始图像和模板图像的傅里叶变换
f_img = np.fft.fft2(img)
f_template = np.fft.fft2(template)

# 计算原始图像和模板图像的共轭傅里叶变换
f_img_conj = np.conj(f_img)
f_template_conj = np.conj(f_template)

# 计算原始图像和模板图像的幅度谱和相位谱
magnitude_img, phase_img = cv2.cartToPolar(f_img.real, f_img.imag)
magnitude_template, phase_template = cv2.cartToPolar(f_template.real, f_template.imag)

# 计算幅度谱的归一化互相关
norm_xcorr = np.fft.ifft2((f_img * f_template_conj) / (magnitude_img * magnitude_template))

# 计算相位谱的归一化互相关
phase_xcorr = np.fft.ifft2((f_img * f_template_conj) / (np.abs(f_img) * np.abs(f_template)))

# 找到归一化互相关的最大值
max_value = np.max(norm_xcorr)

# 找到相位相关的最大值
phase_max_value = np.max(np.abs(phase_xcorr))

# 显示结果
cv2.imshow('Original Image', img)
cv2.imshow('Template Image', template)
cv2.imshow('Normalized Cross Correlation', norm_xcorr.real)
cv2.imshow('Phase Correlation', phase_xcorr.real)
cv2.waitKey()
cv2.destroyAllWindows()

希望对您有所帮助！