samples/cpp/distrans.cpp

一个使用距离变换的示例

#include <opencv2/core/utility.hpp>
#include "opencv2/imgproc.hpp"
#include "opencv2/imgcodecs.hpp"
#include "opencv2/highgui.hpp"
 
#include <stdio.h>
 
using namespace std;
using namespace cv;
 
int maskSize0 = DIST_MASK_5;
int voronoiType = -1;
int edgeThresh = 100;
int distType0 = DIST_L1;
 
// 输出和临时图像
Mat gray;
 
// 阈值跟踪条回调函数
static void onTrackbar( int, void* )
{
    static const Scalar colors[] =
    {
        Scalar(0,0,0),
        Scalar(255,0,0),
        Scalar(255,128,0),
        Scalar(255,255,0),
        Scalar(0,255,0),
        Scalar(0,128,255),
        Scalar(0,255,255),
        Scalar(0,0,255),
        Scalar(255,0,255)
    };
 
    int maskSize = voronoiType >= 0 ? DIST_MASK_5 : maskSize0;
    int distType = voronoiType >= 0 ? DIST_L2 : distType0;
 
    Mat edge = gray >= edgeThresh, dist, labels, dist8u;
 
    if( voronoiType < 0 )
distanceTransform( edge, dist, distType, maskSize );
    else
distanceTransform( edge, dist, labels, distType, maskSize, voronoiType );
 
    if( voronoiType < 0 )
    {
        // 开始“绘制”距离变换结果
dist *= 5000;
pow(dist, 0.5, dist);
 
        Mat dist32s, dist8u1, dist8u2;
 
dist.convertTo(dist32s, CV_32S, 1, 0.5);
dist32s &= Scalar::all(255);
 
dist32s.convertTo(dist8u1, CV_8U, 1, 0);
dist32s *= -1;
 
dist32s += Scalar::all(255);
dist32s.convertTo(dist8u2, CV_8U);
 
        Mat planes[] = {dist8u1, dist8u2, dist8u2};
merge(planes, 3, dist8u);
    }
    else
    {
dist8u.create(labels.size(), CV_8UC3);
        for( int i = 0; i < labels.rows; i++ )
        {
            const int* ll = (const int*)labels.ptr(i);
            const float* dd = (const float*)dist.ptr(i);
            uchar* d = (uchar*)dist8u.ptr(i);
            for( int j = 0; j < labels.cols; j++ )
            {
                int idx = ll[j] == 0 || dd[j] == 0 ? 0 : (ll[j]-1)%8 + 1;
                float scale = 1.f/(1 + dd[j]*dd[j]*0.0004f);
                int b = cvRound(colors[idx][0]*scale);
                int g = cvRound(colors[idx][1]*scale);
                int r = cvRound(colors[idx][2]*scale);
d[j*3] = (uchar)b;
d[j*3+1] = (uchar)g;
d[j*3+2] = (uchar)r;
            }
        }
    }
 
imshow("Distance Map", dist8u );
}
 
static void help(const char** argv)
{
printf("\n程序演示边缘图像之间的距离变换函数的使用。\n"
            "用法：\n"
            "%s [image_name -- 默认图像是 stuff.jpg]\n"
            "\n热键：\n"
            "\tESC - 退出程序\n"
            "\tC - 使用 C/Inf 度量\n"
            "\tL1 - 使用 L1 度量\n"
            "\tL2 - 使用 L2 度量\n"
            "\t3 - 使用 3x3 掩码\n"
            "\t5 - 使用 5x5 掩码\n"
            "\t0 - 使用精确距离变换\n"
            "\tv - 切换到 Voronoi 图模式\n"
            "\tp - 切换到像素级 Voronoi 图模式\n"
            "\tSPACE - 循环遍历所有模式\n\n", argv[0]);
}
 
const char* keys =
{
    "{help h||}{@image |stuff.jpg|输入图像文件}"
};
 
int main( int argc, const char** argv )
{
    CommandLineParser parser(argc, argv, keys);
help(argv);
    if (parser.has("help"))
        return 0;
    string filename = parser.get<string>(0);
gray = imread(samples::findFile(filename), 0);
    if(gray.empty())
    {
printf("无法读取图像文件：%s\n", filename.c_str());
help(argv);
        return -1;
    }
 
    namedWindow("Distance Map", 1);
    createTrackbar("Brightness Threshold", "Distance Map", &edgeThresh, 255, onTrackbar, 0);
 
    for(;;)
    {
        // 调用更新视图
onTrackbar(0, 0);
 
        char c = (char)waitKey(0);
 
        if( c == 27 )
            break;
 
        if( c == 'c' || c == 'C' || c == '1' || c == '2' ||
c == '3' || c == '5' || c == '0' )
voronoiType = -1;
 
        if( c == 'c' || c == 'C' )
distType0 = DIST_C;
        else if( c == '1' )
distType0 = DIST_L1;
        else if( c == '2' )
distType0 = DIST_L2;
        else if( c == '3' )
maskSize0 = DIST_MASK_3;
        else if( c == '5' )
maskSize0 = DIST_MASK_5;
        else if( c == '0' )
maskSize0 = DIST_MASK_PRECISE;
        else if( c == 'v' )
voronoiType = 0;
        else if( c == 'p' )
voronoiType = 1;
        else if( c == ' ' )
        {
            if( voronoiType == 0 )
voronoiType = 1;
            else if( voronoiType == 1 )
            {
voronoiType = -1;
maskSize0 = DIST_MASK_3;
distType0 = DIST_C;
            }
            else if( distType0 == DIST_C )
distType0 = DIST_L1;
            else if( distType0 == DIST_L1 )
distType0 = DIST_L2;
            else if( maskSize0 == DIST_MASK_3 )
maskSize0 = DIST_MASK_5;
            else if( maskSize0 == DIST_MASK_5 )
maskSize0 = DIST_MASK_PRECISE;
            else if( maskSize0 == DIST_MASK_PRECISE )
voronoiType = 0;
        }
    }
 
    return 0;
}