遥感影像自动配准
基于SIFT的光學遙感影像自動匹配及配準(Python版)
基礎SIFT算法只適用于光學遙感影像
代碼如下:
import numpy as np #導入庫
from cv2 import cv2
MIN_MATCH_COUNT = 10 #最少匹配數
path1=r’E:\Vscode\NewData\Paper1Data\image\3\GF21024.tif’ #參考影像路徑
path2=r’E:\Vscode\NewData\Paper1Data\image\3\ZY31024.tif’ #感測影像路徑
img1 = cv2.imread(path1) #基準影像
img2 = cv2.imread(path2) #待配準
if img1.shape!=img2.shape:
img2=cv2.resize(img2,(img1.shape[1],img1.shape[0]),interpolation=cv2.INTER_CUBIC)#重采樣
img3=img1.copy()
img4=img2.copy()
sift = cv2.xfeatures2d.SIFT_create()
kp1, des1 = sift.detectAndCompute(img1,None) #SIFT特征點檢測
kp2, des2 = sift.detectAndCompute(img2,None)
cv2.drawKeypoints(img3,kp1,img3,(0,255,0),2) #繪制特征點
cv2.drawKeypoints(img4,kp2,img4,(0,255,0),2)
index_params = dict(algorithm = 1, trees = 5)
search_params = dict(checks = 50) #FLANN快速匹配
flann = cv2.FlannBasedMatcher(index_params, search_params)
matches = flann.knnMatch(des1,des2,k=2) #KNN匹配
good = [] #存儲粗匹配特征點對
for i,(m,n) in enumerate(matches): #閾值剔除誤匹配點
if m.distance < 0.9*n.distance:
good.append(m)
#單應性矩陣
should_point=[]
src_pts=np.float32([ kp1[m.queryIdx].pt for m in good ]).reshape(-1,1,2)
dst_pts=np.float32([ kp2[m.trainIdx].pt for m in good ]).reshape(-1,1,2)
#單應性變換矩陣RANSAC
H,mask=cv2.findHomography(src_pts,dst_pts,cv2.RANSAC,3.0)
mask2=mask.ravel().tolist()
print(‘內點次數’,mask2.count(1)) #正確點數量
print(‘外點次數’,mask2.count(0))
imgOutH=cv2.warpPerspective(img2,H,(img1.shape[1],img1.shape[0]),flags=cv2.INTER_LINEAR + cv2.WARP_INVERSE_MAP) #配準結果
draw_params = dict(matchColor = (0,0,255),
singlePointColor = (255,0,0),
matchesMask = mask2,
flags = 2)
#繪制匹配點連線圖
imgmatch = cv2.drawMatches(img1,kp1,img2,kp2,good,None,**draw_params)
cv2.namedWindow(“matchimg”,cv2.WINDOW_NORMAL)
cv2.namedWindow(“outimgH”,cv2.WINDOW_NORMAL)
cv2.imshow(“matchimg”,imgmatch)
cv2.imshow(“outimgH”,imgOutH)
cv2.namedWindow(“img3”,cv2.WINDOW_NORMAL)
cv2.namedWindow(“img4”,cv2.WINDOW_NORMAL)
cv2.imshow(“img3”,img3)
cv2.imshow(“img4”,img4)
cv2.waitKey(0)
cv2.destroyAllWindows(0)
下次講述對影像匹配結果進行定量(RMSE,NCM,精度,召回率)和定性分析(棋盤格鑲嵌圖)
有問題的小伙伴+Q:1399212294
總結
- 上一篇: VL53L0X 驱动移植
- 下一篇: 计算机找不到工具怎么办,电脑工具栏声音图