OpenCV骨架提取代码

提取骨架常用的方法有zhang快速细化法，rosenfeld方法等。两种算法的原理可以参考：贾挺猛. 葡萄树冬剪机器人剪枝点定位方法研究[D]. 浙江工业大学, 2012.

这篇文章中有比较详细的解释。此外《数字图像处理》（第三版）-冈萨雷斯，9.5.7和11.1.7两个小节对骨架提取问题进行了比较详细的介绍。下面是实现两种算法的代码。

#define p1 SrcImage.at<uchar>(i, j)  
#define p2 SrcImage.at<uchar>(i - 1, j)  
#define p3 SrcImage.at<uchar>(i - 1, j + 1)  
#define p4 SrcImage.at<uchar>(i, j + 1)  
#define p5 SrcImage.at<uchar>(i + 1, j + 1)  
#define p6 SrcImage.at<uchar>(i + 1, j)  
#define p7 SrcImage.at<uchar>(i + 1, j - 1)  
#define p8 SrcImage.at<uchar>(i, j - 1)  
#define p9 SrcImage.at<uchar>(i - 1, j - 1)  
#define p10 p2

void ZhangFindSkeleton(Mat SrcImage, Mat &DstImage)//输入的目标像素为1，背景像素为0  
{  
    /*0：先对图像的边缘进行处理,避免图片的边缘对骨架提取造成的影响——如果图片四边都为0则没有必要*/  
    for (int j = 0;j < SrcImage.cols;j++)  
    {  
        SrcImage.at<uchar>(0, j) = 0;  
        SrcImage.at<uchar>(SrcImage.rows - 1, j) = 0;  
    }  
    for (int i = 0;i < SrcImage.rows;i++)  
    {  
        SrcImage.at<uchar>(i, 0) = 0;  
        SrcImage.at<uchar>(i, SrcImage.cols - 1) = 0;  
    }  
    /*0:记录所有非零点的坐标*/  
    vector<Point>target;  
    for (int i = 1;i < SrcImage.rows - 1;i++)  
    {  
        for (int j = 1;j < SrcImage.cols - 1;j++)  
        {  
            if (SrcImage.at<uchar>(i, j) == 1)  
            {  
                target.push_back(Point(i, j));  
            }  
        }  
    }  
    //--------------------------Zhang----------------------------------  
    int flag = 1;  
    while (flag)  
    {  
        flag = 0;//清空标志位  
         /*1：检测像素点是否满足Zhang快速细化法步骤1的四个条件*/  
        vector<Point> erase,reserve;  
        for (int k = 0;k < target.size();k++)//只检测已经记录的点  
        {  
            int i = target[k].x, j = target[k].y;  
            //检测条件a 2<=N(p1)<=6  
            int n = p2 + p3 + p4 + p5 + p6 + p7 + p8 + p9;  
            if (n < 2