Programming Computer Vision with Python (学习笔记十)

如今考虑一个全景图拼接的应用场景,假设现有两张图片须要拼接成一张全景图,这两张图片是经过相机右转必定角度拍摄出来的,两张图片有部分取景是重叠的。如何实现拼接?固然这是一个不简单的问题,咱们如今只考虑实现拼接目标的第一步:找出图像中重叠的内容,以及分别在两张图片中的位置。python

以下图所示,左右是两张稍有不一样的图片,但都包含了广州塔,左图红色框中标出了两个感兴趣的点,我指望找出它们在右图的对应位置(即对应点)。
图片描述算法

首先,要肯定检测哪些点,即哪些点是咱们感兴趣的?这可使用Harris角检测(见上篇笔记)方法来获得图像的角点集合,而后经过设置合适的阈值和坐标范围来找出咱们感兴趣的点。有了两个图像的兴趣点集后,又如何能计算出它们的对应关系呢?这就须要解决两个问题:segmentfault

  • 兴趣点如何描述app

  • 兴趣点之间的对应关系如何计算ide

兴趣点描述

兴趣点,也即用Harris角检测出来的结果,它只有坐标和像素值,只有这些信息不足以用于匹配,没法从另外一张图像中查找是否包含这个点。因此须要增长点的表征信息,一种方法是使用围绕点周围一小块的图像来描述这个点,如采用上图中所标记的方式,即:以兴趣点为中心划出一个小矩形,将区域内全部像素值以一贯量进行存储,用这个向量来描述这个兴趣点,那么此向量称为兴趣点描述符(interest point descriptor,下简称IPD)函数

下面实现一个函数,为全部角点生成IPD:spa

def get_desc(image, filtered_coords, wid = 5):
    #image为原图像,filtered_coords为角点的坐标,wid为矩形的“半径”
    desc = []
    for coords in filtered_coords:
        ipd = image[coords[0] - wid : coords[0] + wid + 1,
                coords[1] - wid : coords[1] + wid + 1].flatten()

        if ipd.shape[0] > 0:
            desc.append(ipd)

    return desc

兴趣点相关度

如何肯定左图中的某个兴趣点,对应右图中的某个兴趣点?对应关系,不必定彻底是等价关系,即两个点虽然是对应关系,但它们对应的IPD并不彻底相同。由于咱们这里讨论的找对应点的方法,容许两张图像在亮度、缩放上有必定的区别。因此两个点的对应关系不能用IPD等价来匹配,而是要采用类似度或相关度来计算,相关度越高,它们越多是对应关系。而相关度,可使用现成的数学模型——皮尔逊相关系数(Pearson's r,也被称为皮尔森相关系数r,下简称r系数)来表示。因此,计算两个点的对应关系就转化为计算两个IPD的r系数。rest

r系数被普遍用于度量两个变量之间的相关(线性相关)程度,它是两个变量之间的协方差和标准差的商,一种等价表达式为标准分的均值:
r = 图片描述code

I1和I2为样本集,μ1为I1的平均值,μ2为I2的平均值,σ1为I1的标准差,σ2为I2的标准差,上式计算结果即为r系数,范围为-1到1。 r系数为正且越大,表示I1和I2同时趋向于它们各自的平均值,变化方向一致,相关度越高。系数为0意味着两个变量之间没有线性关系。orm

把两个点对应的IPD代入上述公式的I,可获得两个点的相关程度。因此找两个图像之间兴趣点的对应关系,计算步骤是:

  1. 分别对两个图像应用Harris角检测,获得图像1的兴趣点集1,和图像2的兴趣点集2

  2. 设定IPD的矩形大小,计算全部兴趣点的IPD,获得IPD_SET_1和IPD_SET_2两个集合

  3. 设定r系数的阈值,如0.5,即相关度在[0.5,1]之间咱们才考虑,那么,对IPD_SET_1中指定的某个IPD,计算它与IPD_SET_2中全部IPD的r系数,若最大的r系数落在[0.5,1]区间,则其对应的IPD是最相关的。

下面实现一个IPD匹配函数,传入两个IPD集合,找出全部r系数符合给定阈值的(即认为有对应关系的)IPD:

def match(desc1, desc2, threshold = 0.5):
    n = len(desc1[0])
    count1 = len(desc1)
    count2 = len(desc2)

    d = -np.ones((count1, count2)) #每一个图1的IPD,其对应的力2的IPD下标初始化为-1
    for i in range(count1):
        ipd1 = desc1[i]
        d1 = (ipd1 - np.mean(ipd1)) / np.std(ipd1)
        for j in range(count2):
            ipd2 = desc2[j]
            if ipd1.shape[0] == ipd2.shape[0]: #忽略位于边缘的IPD
                d2 = (ipd2 - np.mean(ipd2)) / np.std(ipd2)
                r = np.sum(d1 * d2) / (n - 1)
                if r > threshold:
                    d[i, j] = r #i为图像1角点坐标, j为符合阈值的图像2角点坐标

    ndx = np.argsort(-d)  #将d的列降序排列,第0列即为r系数最大的
    match_index_array = ndx[:, 0] #只保留第0列
    return match_index_array

上述的函数为图1的每一个IPD,从右边找到最佳的匹配(若是存在),但这还不够,由于这不表明对右边的这个IPD来讲,左边的的这个IPD是它的最佳匹配,因此,若是使用两向匹配,互相认为是最佳的,咱们才认为是对应关系,这样效果会更好一些,双向匹配的函数实现:

def match_twosided(desc1, desc2, threshold = 0.5):
    m_12 = match(desc1, desc2, threshold)
    m_21 = match(desc2, desc1, threshold)

    for i,j in enumerate(m_12):
        if j >= 0 and m_21[j] != i:
            m_12[i] = -1  #非双向匹配的,置为-1,上层应该忽略之

    return m_12

例子

下面代码使用以上的两向匹配方法找出两张图像的对应点,并用白色线链接起来,看一下效果,两张图像是并排显示的:
图片描述

从图中能够看出,两个图像中的广州塔上的关键角点基本能找到对应的位置,但图像的底部即建筑物的角点,与右图的建筑物链接起来,即便它们不是相同的建筑物,这是由于这些角点看起来很像,准确点讲,相关度(r系数)很高,因此被认为是对应点。

从这个例子也能够看出,要准确的找到对象在图像间的对应点,还须要考虑一些因素,来使效果更佳:

  • 为兴趣点定义一个范围,好比上面例子,若是只关注塔尖的兴趣点,得出的效果使人满意

  • 在寻找对应关系时,可限定对应点的y坐标的距离不能超过必定范围(如50个像素,根据实际应用而定),这样能够有效排除一些虽然r系数高,但事实上不对应的点。

代码以下,注意点的疏密能够经过参数微调:

from PIL import Image
import matplotlib.pyplot as plt
import numpy as np
from scipy.ndimage import filters
from skimage.feature import corner_peaks

def harris_eps(im, sigma=3): #harris角检测,见上个笔记
    imx = np.zeros(im.shape)
    filters.gaussian_filter(im, (sigma,sigma), (0,1), imx)
    imy = np.zeros(im.shape)
    filters.gaussian_filter(im, (sigma,sigma), (1,0), imy)
    Wxx = filters.gaussian_filter(imx*imx,sigma)
    Wxy = filters.gaussian_filter(imx*imy,sigma)
    Wyy = filters.gaussian_filter(imy*imy,sigma)
    Wdet = Wxx*Wyy - Wxy**2
    Wtr = Wxx + Wyy
    return Wdet * 2 / (Wtr + 1e-06)

# def get_desc(image, filtered_coords, wid = 5):
# 省略,见上文

# def match_twosided(desc1, desc2, threshold = 0.5):
# 省略,见上文

im1 = np.array(Image.open('tower-left.jpg').convert('L'))
im2 = np.array(Image.open('tower-right.jpg').convert('L'))

coords_1 = corner_peaks(harris_eps(im1, sigma=1), min_distance=3, threshold_abs=0, threshold_rel=0.1)
coords_2 = corner_peaks(harris_eps(im2, sigma=1), min_distance=3, threshold_abs=0, threshold_rel=0.1)

desc1 = get_desc(im1, coords_1, wid=6)
desc2 = get_desc(im2, coords_2, wid=6)

match_index_array = match_twosided(desc1, desc2, threshold=0.5)

im3 = np.concatenate((im1, im2), axis=1) #将两个图像左右合并成一个,以便显示
plt.gray()
plt.imshow(im3)

for ipd_index_1,ipd_index_2 in enumerate(match_index_array):
    if ipd_index_2 != -1:
        x = [coords_1[ipd_index_1][4], coords_2[ipd_index_2][5] + im1.shape[1]]
        y = [coords_1[ipd_index_1][0], coords_2[ipd_index_2][0]]
        if np.abs(y[0] - y[1]) < 50: #这里限制了对应点之间的y坐标距离
            plt.plot(x, y, 'w', alpha=0.5) #链接两个对应点

plt.plot(coords_1[:, 1], coords_1[:, 0], '+r', markersize=5) #画图1角点坐标
plt.plot(coords_2[:, 1] + im1.shape[1], coords_2[:, 0], '+r', markersize=5)  #画图2角点坐标
plt.axis('off')
plt.show()

小结

从实例中能够看到,本文使用的描述点的和匹配的方法,存在误配的状况,矩形大小的设置也会影响匹配的结果,并且它也不适用于在图像被旋转和缩放的状况下使用,近年,关于这方面的研究也在不断取得进步,下一笔记将介绍一种称为尺度不变特征转换(Scale-invariant feature transform 或 SIFT)的算法,此算法应用很是广。

你还能够查看个人其它笔记

参考资料

wiki 皮尔逊积矩相关系数

相关文章
相关标签/搜索