霍夫变换
前言
今天群里有人问到一个图像的问题,但本质上是一个基本最小二乘问题,涉及到霍夫变换(Hough Transform),用到了就顺便总结一下。
内容为自己的学习记录,其中多有参考他人,最后一并给出链接。
一、霍夫变换(Hough)
A-基本原理
一条直线可由两个点A=(X1,Y1)和B=(X2,Y2)确定(笛卡尔坐标)
另一方面,也可以写成关于(k,q)的函数表达式(霍夫空间):
对应的变换可以通过图形直观表示:
变换后的空间成为霍夫空间。即:笛卡尔坐标系中一条直线,对应霍夫空间的一个点。
反过来同样成立(霍夫空间的一条直线,对应笛卡尔坐标系的一个点):
再来看看A、B两个点,对应霍夫空间的情形:
一步步来,再看一下三个点共线的情况:
可以看出如果笛卡尔坐标系的点共线,这些点在霍夫空间对应的直线交于一点:这也是必然,共线只有一种取值可能。
如果不止一条直线呢?再看看多个点的情况(有两条直线):
其实(3,2)与(4,1)也可以组成直线,只不过它有两个点确定,而图中A、B两点是由三条直线汇成,这也是霍夫变换的后处理的基本方式:选择由尽可能多直线汇成的点。
看看,霍夫空间:选择由三条交汇直线确定的点(中间图),对应的笛卡尔坐标系的直线(右图)。
到这里问题似乎解决了,已经完成了霍夫变换的求解,但是如果像下图这种情况呢?
k=∞是不方便表示的,而且q怎么取值呢,这样不是办法。因此考虑将笛卡尔坐标系换为:极坐标表示。
在极坐标系下,其实是一样的:极坐标的点→霍夫空间的直线,只不过霍夫空间不再是[k,q]的参数,而是的参数,给出对比图:
是不是就一目了然了?
给出霍夫变换的算法步骤:
对应code:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
|
function [ Hough, theta_range, rho_range ] = naiveHough(I) %NAIVEHOUGH Peforms the Hough transform in a straightforward way. % [rows, cols] = size (I); theta_maximum = 90; rho_maximum = floor ( sqrt (rows^2 + cols^2)) - 1; theta_range = -theta_maximum:theta_maximum - 1; rho_range = -rho_maximum:rho_maximum; Hough = zeros ( length (rho_range), length (theta_range)); for row = 1:rows for col = 1:cols if I(row, col) > 0 %only find: pixel > 0 x = col - 1; y = row - 1; for theta = theta_range rho = round ((x * cosd (theta)) + (y * sind (theta))); %approximate rho_index = rho + rho_maximum + 1; theta_index = theta + theta_maximum + 1; Hough(rho_index, theta_index) = Hough(rho_index, theta_index) + 1; end end end end |
其实本质上就是:
交点怎么求解呢?细化成坐标形式,取整后将交点对应的坐标进行累加,最后找到数值最大的点就是求解的,也就求解出了直线。
B-理论应用
这里给出MATLAB自带的一个应用,主要是对一幅图像进行直线检验,原图像为:
首先是对其进行边缘检测:
边缘检测后并二值化,就可以通过找非零点的坐标确定数据点。从而对数据点进行霍夫变换。对应映射到霍夫空间的结果为:
找出其中数值较大的一些点,通常可以给定一个阈值,Threshold一下。
这就完成了霍夫变换的整个过程。这个时候求解出来了其实就是多条直线的斜率k以及截距q,通常会根据直线的特性进一步判断,从而将直线变为线段:
不过这一步更类似后处理,其实已经不是霍夫变换本身的特性了。
给出对应的代码:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
|
clc ; clear all ; close all ; I = imread ( 'circuit.tif' ); rotI = imrotate(I,40, 'crop' ); subplot 221 fig1 = imshow(rotI); BW = edge(rotI, 'canny' ); title ( '原图像' ); subplot 222 imshow(BW); [H,theta,rho] = hough(BW); title ( '图像边缘检测' ); subplot 223 imshow(imadjust(mat2gray(H)),[], 'XData' ,theta, 'YData' ,rho,... 'InitialMagnification' , 'fit' ); xlabel ( '\theta (degrees)' ), ylabel ( '\rho' ); axis on, axis normal, hold on; colormap (hot) P = houghpeaks(H,5, 'threshold' , ceil (0.7* max (H(:)))); x = theta(P(:,2)); y = rho(P(:,1)); plot (x,y, 's' , 'color' , 'black' ); lines = houghlines(BW,theta,rho,P, 'FillGap' ,5, 'MinLength' ,7); title ( 'Hough空间' ); subplot 224, imshow(rotI), hold on max_len = 0; for k = 1: length (lines) xy = [lines(k).point1; lines(k).point2]; plot (xy(:,1),xy(:,2), 'LineWidth' ,2, 'Color' , 'green' ); % Plot beginnings and ends of lines plot (xy(1,1),xy(1,2), 'x' , 'LineWidth' ,2, 'Color' , 'yellow' ); plot (xy(2,1),xy(2,2), 'x' , 'LineWidth' ,2, 'Color' , 'red' ); % Determine the endpoints of the longest line segment len = norm (lines(k).point1 - lines(k).point2); if ( len > max_len) max_len = len; xy_long = xy; end end % highlight the longest line segment plot (xy_long(:,1),xy_long(:,2), 'LineWidth' ,2, 'Color' , 'red' ); title ( '直线检测' ); |
对比自带的Hough与编写的Hough:
效果还是比较接近的。
看到Stackoverflow上的一个答案,觉得很好,收藏一下:
[2017-04-25 10:25:37]申请搜狐自媒体平台