我正在尝试为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;
}
它完成了这项工作。但它很慢。有没有更快更有效的方法呢?
答案 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默认计算包含整数值的标签图像。