更快速地获取给定点的blob

时间:2018-03-09 12:34:45

标签: c++ opencv image-processing

我正在尝试为matlab的bwselect创建一个等效函数。所以,我想显示blob(包含我将提供的点)并掩盖其余部分。 这是我尝试过的。

cv::Mat bwselect(cv::Mat matImg, int x, int y)
{
    cv::Mat img_labels, stats, centroids, mask;
    if (matImg.data)
    {
        int numOfLables = connectedComponentsWithStats(matImg, img_labels, stats, centroids, 8, CV_32S);

        if (numOfLables > 1)
        {
            for (int i = 1; i < numOfLables; i++)
            {
                mask = cv::Mat::zeros(img_labels.size(), CV_8UC1);
                mask = mask | (img_labels == i);
                if (mask.at<uchar>(y, x) > 0)
                {
                    break;
                }

            }
        }
    }
    return mask;
}

它完成了这项工作。但它很慢。有没有更快更有效的方法呢?

1 个答案:

答案 0 :(得分:2)

如果输入图像很大并且它包含许多对象,那么可能会出现瓶颈,因为您要多次分配/取消分配大的掩码缓冲区。

此外,如果您多次调用此函数,最好只调用connectedComponentsWithStats一次,然后将其结果用作函数的附加输入。

我建议您替换整个循环

for (int i = 1; i < numOfLabels; i++){/*...*/}

用这个

 // img_labels data type is CV_32S
 int label_at_pos = img_labels.at<int>(y, x); 

 if (label_at_pos > 0)
 {
    // create mask here and return it
 }

编辑:我对上面的代码示例进行了更正。 connectedComponentsWithStats默认计算包含整数值的标签图像。