python ransac_在OpenCV3中进行特征点筛选与优化-Python版

在前面的记录中对使用OpenCV2进行特征匹配的步骤做了一个简单的介绍,详见:https://www.jianshu.com/p/5083f8d75439,使用了ORB提取特征点,但是匹配出的结果是非常粗糙的,结果显然包含了很多错误的匹配,这就需要使用一些方法来过滤掉这些错误的匹配。因此继续在这里记录对特征点进行进一步筛选的过程,同时,也对几个关键的计算流程做个更详细的说明,方便将来参考。

特征点匹配

OpenCV里面的二维特征匹配,有两种常用的方法,

Brute Force Matcher

Flann based Matcher

继承于DescriptorMatcher类,分别对应BFMatcher和FlannBasedMatcher。

二者的区别: BFMatcher总是尝试所有可能的匹配,从而使得它总能够找到最佳匹配。这也是Brute Force(暴力法)的原始含义。方法是计算某一个特征点描述子与其他所有特征点描述子之间的距离,然后将得到的距离进行排序,取距离最近的一个作为匹配点。 FlannBasedMatcher中FLANN的含义是Fast Library forApproximate Nearest Neighbors,它是一种近似法,算法更快但是找到的是最近邻近似匹配,当我们需要找到一个相对好的匹配但是不需要最佳匹配的时候可以用FlannBasedMatcher。当然也可以通过调整FlannBasedMatcher的参数来提高匹配的精度或者提高算法速度,但是相应地算法速度或者算法精度会受到影响。

此外,使用特征提取过程得到的特征描述符(descriptor)数据类型有的是float类型的,ORB和BRIEF特征描述子只能使用BruteForce匹配法。

具体matching的方法

match: 给定查询集合中的每个特征描述子,寻找最佳匹配,返回值按距离升序排列。

knnMatch:给定查询集合中的每个特征描述子,寻找k个最佳匹配

radiusMatch:在特定范围内寻找特征描述子,返回特定距离内的匹配

特征点筛选

通过上述过程得到了raw_matches之后,接下来对其进行筛选。

交叉验证(Cross-match filter)

这种方法只针对BFMatcher, 就是将BFMatcher的最后一个参数,交叉验证声明为true,即matcher = cv2.BFMatcher(cv2.NORM_HAMMING,crossCheck=True),如果图1的一个特征点和图2的一个特征点相匹配,则进行一个相反的检查,即从图2上的特征点进行匹配图1的特征点,如果相互之间都匹配成功,才认为是一个好的匹配。

cross-match filter 结果如图(似乎还不够好):

long-cross-match.jpg

knnMatch

knnMatch返回K个好的匹配,k可以自行指定。这里指定k=2,raw_matches = matcher.knnMatch(desc1, desc2,2) ,然后每个match得到两个最接近的descriptor,再计算最接近距离和次接近距离之间的比值,当比值大于某个设定的值时,才作为最终的match。

knnMatch结果如图:

long-knn-match.jpg

RANSAC

为了进一步提升精度,还可以采用随机采样一致性(RANSAC)来过滤错误的匹配,该方法是利用匹配点计算两图像之间的单应矩阵,然后利用重投影误差来判定某一个匹配是否是正确的匹配。OpenCV中封装了求解单应矩阵的方法 findHomography ,可以为该方法设定一个重投影误差的阈值,可以得到一个向量mask来指定那些是符合重投影误差的匹配点对,用来过滤掉错误的匹配。

RANSAC结果如图,左边是ORB后未筛选的结果,右边是筛选优化后的:

ransac-1.jpg

ransac-2.jpg

可以看出,优化的作用还是很明显的。最后,贴项目代码如下:

"""

Created on Fri Jan 12 17:59:06 2018

Ransac test

@author:wanqingfeng

"""

import numpy as np

import cv2

img1 = cv2.imread("pic/face1.jpg",0)

img2 = cv2.imread("pic/face2.jpg",0)

detector = cv2.ORB_create()

flann_params= dict(algorithm = 6,

table_number = 6, # 12

key_size = 12, # 20

multi_probe_level = 1) #2

matcher = cv2.BFMatcher(cv2.NORM_HAMMING,crossCheck=False)

kp1,desc1 = detector.detectAndCompute(img1,None)

kp2,desc2 = detector.detectAndCompute(img2,None)

raw_matches = matcher.knnMatch(desc1, desc2,2) #2

good = []

for m,n in raw_matches:

if m.distance < 0.7*n.distance:

good.append(m)

if len(good)>10:

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)

M, mask = cv2.findHomography(src_pts, dst_pts, cv2.RANSAC,5.0)

matchesMask = mask.ravel().tolist()

else:

print("Not enough matches are found - %d/%d" % (len(good),10))

matchesMask = None

draw_params = dict(matchColor = (0,255,0), # draw matches in green color

singlePointColor = (0,0,255),

matchesMask = matchesMask,

flags = 2)# draw only inliers

vis = cv2.drawMatches(img1,kp1,img2,kp2,good,None,**draw_params)

cv2.imshow("", vis)

cv2.imwrite("pic/face_brisk_bf_ransac_1519.jpg", vis)

cv2.waitKey()

cv2.destroyAllWindows()

  • 1
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值