一键找出图像中物体的角点（论文复现）

wei_shuo 2024-10-23

一键找出图像中物体的角点（论文复现）

本文所涉及所有资源均在传知代码平台可获取

一键找出图像中物体的角点（论文复现）

概述
算法原理
演示效果
核心逻辑
使用方式

概述

本文复现论文A COMBINED CORNER AND EDGE DETECTOR中提出的图像中的物体角点检测算法,也称Harris算法。原文连接该论文所提出的角点检测算法是计算机视觉领域的经典算法，至今仍在计算机视觉领域有着广泛的应用。该方法对算力几乎没有任何要求，依据图像邻域内灰度值的分布特点来对图像中各区域的角点进行判断。不像深度学习需要依靠大量数据的训练且对数据集分布有依赖性，所以时至今日依然在工业界有着很高的应用价值。

算法原理

正如边缘在灰度图像中在某一个方向上会出现明显的灰度变化，角点在图像中往往是在两个方向上都出现明显的灰度变化。

在这里插入图片描述

如在此图中，红点标注处的角点在两个箭头方向的垂直方向上都有明显的灰度变化，两个红色箭头标注出了它的两个灰度变换边缘。

假设我们现在要检测的图片为II,在图像处理中，能够充分体现图像灰度变化特征的是它的方向导数，IxI**x与IyI**y,分别代表图像II的水平方向导数与竖直方向导数

在这里插入图片描述

其中k是一个经验参数，本算法中取0.02，当然也可以根据具体情况进行调整，通常来说k越大，算法对角点的判别就越严格，会有更少的点被判别为角点；k越小，算法的判别就越松一些，会有更多的点被判别为角点。

得到R图后，将R图中最大值RmaxRma**x的0.2倍作为阈值。将R图中小于0.2Rmax0.2Rma**x的像素坐标判别为是角点坐标，其余的则为非角点坐标，最终得到我们的结果图

演示效果

在这里插入图片描述

核心逻辑

def HarrisCornerDetector(image,k=0.02):
# 读取图像
    # 将图像转换为numpy数组
    I= image

    I=np.array(I)

    sobel_x = np.array([[0, 0, 0], [-0.5, 0, 0.5], [0, 0, 0]])
    sobel_y=np.array([[0,0.5,0],[0,0,0],[0,-0.5,0]])

    # 对图像进行水平方向的卷积
    I_x = convolve2d(I, sobel_x, mode="same", boundary="symm")
    I_y=convolve2d(I,sobel_y,mode='same',boundary='symm')


    I2_x=I_x**2
    I2_y=I_y**2
    Ix_Iy=I_x*I_y

    sobel=np.array([[1,1,1],[1,1,1],[1,1,1]])

    A=convolve2d(I2_x,sobel,mode="same",boundary="symm")
    B=convolve2d(Ix_Iy,sobel,mode="same",boundary="symm")
    C=convolve2d(I2_y,sobel,mode="same",boundary="symm")

    R=(A*C-B**2)-k*(A+C)**2

    R_max=np.max(R[1:-1,1:-1])

    C=R

    C[C<=(0.2*R_max)]=0
    C[C>(0.2*R_max)]=1

    return C

使用方式

基础镜像：python:3.9.7

配置环境

解压后进入HarrisCornerDetector项目路径下
运行以下命令：
pip install -r requirements.txt
直接运行
python main.py
即可看到视频中示例图片的角点检测效果

更换图片运行

修改main.py文件里的默认图片路径

将"data/rice.png"更改为你自己的图片路径即可。

同时，我也找了第二张示例图片放在了data路径下，将"data/rice.png"更换为"data/1.webp",就可以看到演示图片中棋盘的角点检测效果