随机抽样一致性算法(RANSAC)示例及源代码

作者:王先荣

    大约在两年前翻译了《随机抽样一致性算法RANSAC》,在文章的最后承诺写该算法的C#示例程序。可惜光阴似箭,转眼许久才写出来,实在抱歉。本文将使用随机抽样一致性算法来来检测直线和圆,并提供源代码下载。

一、RANSAC检测流程

    在这里复述下RANSAC的检测流程,详细的过程见上一篇翻译文章:

    RANSAC算法的输入是一组观测数据,一个可以解释或者适应于观测数据的参数化模型,一些可信的参数。

    RANSAC通过反复选择数据中的一组随机子集来达成目标。被选取的子集被假设为局内点,并用下述方法进行验证:
    1.有一个模型适应于假设的局内点,即所有的未知参数都能从假设的局内点计算得出。
    2.用1中得到的模型去测试所有的其它数据,如果某个点适用于估计的模型,认为它也是局内点。
    3.如果有足够多的点被归类为假设的局内点,那么估计的模型就足够合理。
    4.然后,用所有假设的局内点去重新估计模型,因为它仅仅被初始的假设局内点估计过。
    5.最后,通过估计局内点与模型的错误率来评估模型。
    这个过程被重复执行固定的次数,每次产生的模型要么因为局内点太少而被舍弃,要么因为比现有的模型更好而被选用。

二、得到观测数据

    我们没有实验(测试)数据,这里用手工输入的数据来替代——记录您在PictureBox中的点击坐标,作为观测数据。

        /// <summary>
        /// 得到样本点
        /// </summary>
        /// <param name="sender"></param>
        /// <param name="e"></param>
        private void pbSample_Click(object sender, EventArgs e)
        {
            MouseEventArgs me=(MouseEventArgs)e;
            txtRandomPoints.Text += string.Format("({0},{1}),", me.X, me.Y);
            DrawPoint(new Point(me.X, me.Y));
        }
得到样本点

 

三、检测直线

    3.1 直线的相关知识

    (1)平面上的任意两点可以确定一条直线;

    (2)直线的通用数学表达形式为:ax+by+c=0。这种表达形式有三个未知数,需要提供三个点才能解出a,b,c三个参数。由于随机选择的三个点不一定在一条直线上,所以程序中放弃这种方式。

    (3)直线可以用y=ax+b及x=c这两个式子来表示。这两种形式只有一个或者两个未知数,只需两个点就能解出a,b,c三个参数。随机选择的两个点即可得到直线,我们采用这种形式。

    3.2 直线类

    直线类(Line)封装了跟直线相关的一些属性及方法,列表如下:

    (1)属性

    A——y=ax+b中的a

    B——y=ax+b中的b

    C——x=c中的c

    (2)构造函数

    public Line(PointF p1, PointF p2)

    提供两个点p1及p2,计算出直线的属性A,B,C。

    (3)方法

    GetDistance——获取点到直线之间的距离;

    GetY——根据x坐标,获取直线上点的y坐标;

    ToString——获取直线的方程式。

using System;
using System.Collections.Generic;
using System.Text;
using System.Drawing;
using System.Drawing.Drawing2D;

namespace Ransac
{
    /// <summary>
    /// 直线类:采用y=a*x+b或者x=c的形式表示直线。
    /// </summary>
    public class Line
    {
        /// <summary>
        /// y=ax+b中的a
        /// </summary>
        public double A { get; set; }
        /// <summary>
        /// y=ax+b中的b
        /// </summary>
        public double B { get; set; }
        /// <summary>
        /// x=c中的c
        /// </summary>
        public double C { get; set; }
        
        /// <summary>
        /// 构造函数(如果直线为y=ax+b形式,则C为Nan;如果直线为x=c形式,则A和B为Nan)
        /// </summary>
        /// <param name="a"></param>
        /// <param name="b"></param>
        /// <param name="c"></param>
        public Line(double a, double b, double c)
        {
            if (!((double.IsNaN(a) && double.IsNaN(b) && !double.IsNaN(c)) || (!double.IsNaN(a) && !double.IsNaN(b) && double.IsNaN(c))))
                throw new ArgumentException("参数错误,无效的直线参数。");
            A = a;
            B = b;
            C = c;
        }

        /// <summary>
        /// 构造函数,由两个点确定直线
        /// </summary>
        /// <param name="p1"></param>
        /// <param name="p2"></param>
        public Line(PointF p1, PointF p2)
        {
            if (p1.X == p2.X)
            {
                A = double.NaN;
                B = double.NaN;
                C = p1.X;
            }
            else
            {
                A = 1d * (p1.Y - p2.Y) / (p1.X - p2.X);
                B = p1.Y - A * p1.X;
                C = double.NaN;
            }
        }

        /// <summary>
        /// 构造函数,由两个点确定直线
        /// </summary>
        /// <param name="p1"></param>
        /// <param name="p2"></param>
        public Line(Point p1, Point p2) :
            this(new PointF(p1.X, p1.Y), new PointF(p2.X, p2.Y))
        {
        }

        /// <summary>
        /// 生成一条随机的直线
        /// </summary>
        /// <returns></returns>
        public static Line GetRandomLine()
        {
            Random random = new Random();
            int a = random.Next(-10, 10);
            int b = random.Next(-10, 10);
            return new Line(a, b, double.NaN);
        }

        /// <summary>
        /// 获取点到直线的距离
        /// </summary>
        /// <param name="p"></param>
        /// <returns>返回点到直线的距离;如果直线通过点,返回0。</returns>
        public double GetDistance(Point p)
        {
            return GetDistance(new PointF(p.X, p.Y));
        }

        /// <summary>
        /// 获取点到直线的距离
        /// </summary>
        /// <param name="p"></param>
        /// <returns>返回点到直线的距离;如果直线通过点,返回0。</returns>
        public double GetDistance(PointF p)
        {
            double d = 0d;
            if (double.IsNaN(C))
            {
                //y=ax+b相当于ax-y+b=0
                d = Math.Abs(1d * (A * p.X - p.Y + B) / Math.Sqrt(A * A + 1));
            }
            else
            {
                d = Math.Abs(C - p.X);
            }
            return d;
        }

        /// <summary>
        /// 根据x坐标,得到直线上点的y坐标 
        /// </summary>
        /// <param name="x"></param>
        /// <returns></returns>
        public double GetY(double x)
        {
            double y;
            if (double.IsNaN(C))
                y = A * x + B;
            else
                y = double.NaN;
            return y;
        }

        /// <summary>
        /// 返回直线方程
        /// </summary>
        /// <returns></returns>
        public override string ToString()
        {
            string formula = "";
            if (double.IsNaN(C))
                formula = string.Format("y={0}{1}", A != 0 ? string.Format("{0:F02}x", A) : "", B != 0 ? (B > 0 ? string.Format("+{0:F02}", B) : string.Format("{0:F02}", B)) : "");
            else
                formula = string.Format("x={0:F02}", C);
            return formula;
        }
    }
}
Line类

 

    3.3 检测直线的过程

    (1)随机从观测点中选择两个点,得到通过该点的直线;

    (2)用(1)中的直线去测试其他观测点,由点到直线的距离确定观测点是否为局内点或者局外点;

    (3)如果局内点足够多,并且局内点多于原有“最佳”直线的局内点,那么将这次迭代的直线设为“最佳”直线;

    (4)重复(1)~(3)步直到找到最佳直线。

    细心的您估计已经发现我省略了标准RANSAC检测过程中重新估计模型的步骤,我是故意的,我觉得麻烦且没什么用处,所以咔嚓了,O(∩_∩)O~。

/// <summary>
        /// 尝试获取直线
        /// </summary>
        /// <param name="sender"></param>
        /// <param name="e"></param>
        private void btnGetLine_Click(object sender, EventArgs e)
        {
            //用RANSAC方法获取最佳直线
            points = GetSamplePoints();
            Line bestLine = null;           //最佳直线
            double bestInliersCount = 0;    //最佳模型的局内点数目
            Random random = new Random();
            for (int idx = 0; idx < nudIterCount.Value; idx++)
            {
                int idx1, idx2;
                GetRandomInliersPoints(random, out idx1, out idx2);
                int inliersCount = 2;
                Line line = new Line(points[idx1], points[idx2]);
                for (int i = 0; i < points.Count; i++)
                {
                    if (i != idx1 && i != idx2)
                    {
                        if (line.GetDistance(points[i]) <= (double)nudMinDistance.Value)
                            inliersCount++;
                    }
                }
                if (inliersCount >= nudMinPointCount.Value)
                {
                    if (inliersCount > bestInliersCount)
                    {
                        bestLine = line;
                        bestInliersCount = inliersCount;
                    }
                }
            }
            //显示最佳直线
            if (bestLine != null)
            {
                lblFormula.Text = string.Format("方程:{0}\r\nA:{1}\r\nB:{2}\r\nC:{3}\r\n局内点数目:{4}",
                    bestLine.ToString(), bestLine.A, bestLine.B, bestLine.C, bestInliersCount);
                DrawLine(bestLine);
            }
            else
                lblFormula.Text = "没有获取到最佳直线。";
        }
获取直线

 

四、检测圆

    4.1 圆的相关知识

    (1)平面内不在同一直线上的三个点可以确定一个圆;

    (2)圆的数学表达形式为:(x-a)2+(y-b)2=r2

    其中,(a,b)为圆心,r为半径。

    4.2 圆类

    圆类(Circle)封装了跟圆有关的属性及方法,列表如下:

    (1)属性

    A——圆心的x坐标

    B——圆心的y坐标

     R——圆的半径

    (2)构造函数

    public Circle(PointF p1, PointF p2, PointF p3)

    提供三个点p1,p2和p3,计算出圆的属性A,B,R。

    (3)方法

    GetDistance——获取点到圆(周)之间的距离,表示点接近或者远离圆;

    ToString——获取圆的方程式。

using System;
using System.Collections.Generic;
using System.Text;
using System.Drawing;
using System.Drawing.Drawing2D;

namespace Ransac
{
    /// <summary>
    /// 圆类:用(x-a)**2+(y-b)**2=r**2形式表示
    /// </summary>b
    public class Circle
    {
        /// <summary>
        /// 圆心的X坐标
        /// </summary>
        public double A { get; set; }
        /// <summary>
        /// 圆心的Y坐标
        /// </summary>
        public double B { get; set; }
        /// <summary>
        /// 半径
        /// </summary>
        public double R { get; set; }

        /// <summary>
        /// 构造函数,提供圆心和半径。
        /// </summary>
        /// <param name="a"></param>
        /// <param name="b"></param>
        /// <param name="r"></param>
        public Circle(double a, double b, double r)
        {
            A = a;
            B = b;
            if (r < 0)
                throw new ArgumentOutOfRangeException("r", "圆的半径必须大于0。");
            R = r;
        }

        /// <summary>
        /// 构造函数,提供三个点。
        /// 该算法来自csdn论坛,帖子地址是:http://bbs.csdn.net/topics/50383586,在此感谢5楼的privet。
        /// </summary>
        /// <param name="p1"></param>
        /// <param name="p2"></param>
        /// <param name="p3"></param>
        public Circle(PointF p1, PointF p2, PointF p3)
        {
            float xMove = p1.X;
            float yMove = p1.Y;
            p1.X = 0;
            p1.Y = 0;
            p2.X = p2.X - xMove;
            p2.Y = p2.Y - yMove;
            p3.X = p3.X - xMove;
            p3.Y = p3.Y - yMove;
            float x1 = p2.X, y1 = p2.Y, x2 = p3.X, y2 = p3.Y;
            double m = 2.0 * (x1 * y2 - y1 * x2);
            if (m == 0)
                throw new ArgumentException("参数错误,提供的三点不能构成圆。");
            double x0 = (x1 * x1 * y2 - x2 * x2 * y1 + y1 * y2 * (y1 - y2)) / m;
            double y0 = (x1 * x2 * (x2 - x1) - y1 * y1 * x2 + x1 * y2 * y2) / m;
            R = Math.Sqrt(x0 * x0 + y0 * y0);
            A = x0 + xMove;
            B = y0 + yMove;
        }

        /// <summary>
        /// 构造函数,提供三个点。
        /// </summary>
        /// <param name="p1"></param>
        /// <param name="p2"></param>
        /// <param name="p3"></param>
        public Circle(Point p1, Point p2, Point p3) :
            this(new PointF(p1.X, p1.Y), new PointF(p2.X, p2.Y), new PointF(p3.X, p3.Y))
        {
        }

        /// <summary>
        /// 获取点到圆的距离(圆周,不是圆心)
        /// </summary>
        /// <param name="p"></param>
        /// <returns></returns>
        public double GetDistance(PointF p)
        {
            return Math.Abs(R - Math.Sqrt(Math.Pow(p.X - A, 2) + Math.Pow(p.Y - B, 2)));
        }

        /// <summary>
        /// 返回圆方程
        /// </summary>
        /// <returns></returns>
        public override string ToString()
        {
            return string.Format("{0}**2+{1}**2={2}",
                A == 0 ? "x" : (A > 0 ? string.Format("(x-{0:F02})", A) : string.Format("(x+{0:F02})", Math.Abs(A))),
                B == 0 ? "y" : (B > 0 ? string.Format("(y-{0:F02})", B) : string.Format("(y+{0:F02})", Math.Abs(B))),
                R == 0 ? "0" : string.Format("{0:F02}**2", Math.Abs(R)));
        }
    }
}
Circle类

 

    3.3 检测圆的过程

    (1)随机从观测点中选择三个点,尝试得到通过这三个点的圆;

    (2)用(1)中的圆去测试其他观测点,由点到圆的距离确定观测点是否为局内点或者局外点;

    (3)如果局内点足够多,并且局内点多于原有“最佳”圆的局内点,那么将这次迭代的圆设为“最佳”圆;

    (4)重复(1)~(3)步直到找到最佳圆。

/// <summary>
        /// 尝试获取圆
        /// </summary>
        /// <param name="sender"></param>
        /// <param name="e"></param>
        private void btnGetCircle_Click(object sender, EventArgs e)
        {
            //用RANSAC方法获取最佳直线
            points = GetSamplePoints();
            Circle bestCircle = null;       //最佳圆
            double bestInliersCount = 0;    //最佳模型的局内点数目
            Random random = new Random();
            for (int idx = 0; idx < nudIterCount.Value; idx++)
            {
                int idx1, idx2, idx3;
                GetRandomInliersPoints(random, out idx1, out idx2, out idx3);
                int inliersCount = 3;
                Circle circle;
                try
                {
                    circle = new Circle(points[idx1], points[idx2], points[idx3]);
                }
                catch
                {
                    continue;
                }
                for (int i = 0; i < points.Count; i++)
                {
                    if (i != idx1 && i != idx2 && i!=idx3)
                    {
                        if (circle.GetDistance(points[i]) <= (double)nudMinDistance.Value)
                            inliersCount++;
                    }
                }
                if (inliersCount >= nudMinPointCount.Value)
                {
                    if (inliersCount > bestInliersCount)
                    {
                        bestCircle = circle;
                        bestInliersCount = inliersCount;
                    }
                }
            }
            //显示最佳圆
            if (bestCircle != null)
            {
                lblFormula.Text = string.Format("方程:{0}\r\nA:{1}\r\nB:{2}\r\nR:{3}\r\n局内点数目:{4}",
                    bestCircle.ToString(), bestCircle.A, bestCircle.B, bestCircle.R, bestInliersCount);
                DrawCircle(bestCircle);
            }
            else
                lblFormula.Text = "没有获取到最佳圆。";
        }
获取圆

 

五、本文源代码

    点击这里下载本文源代码

    感谢您阅读本文,希望对您有所帮助。

posted @ 2014-02-03 08:25  Wuya  阅读(21974)  评论(11编辑  收藏  举报