1. 程式人生 > >halcon之最小二乘擬合直線

halcon之最小二乘擬合直線

如果不瞭解最小二乘演算法 請先閱讀: Least squares的演算法細節原理https://en.wikipedia.org/wiki/Least_squares

通常在halcon中擬合直線會用houghline或者 fitline。本文提供一種新的選擇,用halcon的矩陣操作實現最小二乘擬合直線

 

1

2

3

4

5

6

7

8

9

首先隨機生成一組資料

Mx:=[100:10:500]

tuple_length(Mx,len)

tuple_gen_const(len,5,r)

Ma:=2

Mb:=40

tuple_rand(len , noise)

My:= Ma *Mx + Mb*noise

gen_circle(ContCircle, My, Mx, r)

 

1

2

3

4

5

6

7

接下來用矩陣進行最小二乘求解

* y = ax + b

* y1 = ax1 + b

* y2 = ax2 + b

* ... .......

* yn = ax + b

1

 

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

create_matrix(len,1,My,y)

create_matrix(len,2,1,x)

set_value_matrix(x, [0:len-1], gen_tuple_const(len, 0),Mx)

  

* XT 代表X的轉置 inv(*)代表*的逆

* x beta = y

* xT x beta = xT y

* beta = inv( xT x) xT y

  

mult_matrix(x,x,'ATB',xtx)

mult_matrix(x,y,'ATB',xty)

 

 

invert_matrix(xtx,'general', 0, invxtx)

  

mult_matrix(invxtx,xty,'AB', beta)

  

get_full_matrix(beta, Values)

  

Newy:=Values[0] * [10,800] + Values[1]

  

gen_contour_polygon_xld(Contour, Newy, [10,800])