实践教程｜基于OpenCV提取特定区域方法汇总

会员服务 ·

实践教程｜基于OpenCV提取特定区域方法汇总

2021 年 12 月 5 日 极市平台

↑ 点击蓝字关注极市平台

来源丨小白学视觉

编辑丨极市平台

极市导读

本文探究如何使用OpenCV和Python从图像中提取感兴趣区域（ROI）。 >>加入极市CV技术交流群，走在计算机视觉的最前沿

今天我们将一起探究如何使用OpenCV和Python从图像中提取感兴趣区域（ROI）。

在之间的文章中，我们完成了图像边缘提取，例如从台球桌中提取桌边。使用了简单的OpenCV函数即可完成这项任务，例如inRange、findContours、boundingRect、minAreaRect、minEnclosingCircle、circle、HoughLines、line等，都可以。

今天我们的任务是从包含患者大脑活动快照的图像中提取所需的片段。之后可以将该提取的过程应用于其他程序中，例如诊断健康与否的机器学习模型。

因此，让我们从查看输入图像开始。这是由神经科学领域的医疗仪器生成的典型报告，该仪器使用传感器检测来自患者大脑的信号并将其显示为彩色地图。通常，有四张图片，所有图片都描绘了某个特征并一起分析以进行诊断。

从上面的图像中，我们只想提取与四个地图（头部扫描）相对应的区域，而将其他所有内容都排除在外。因此，让我们开始吧。

第一步是检测我们要提取的片段的边缘。这是一个多步骤过程，如下所述：

1. 使用“ cvtColor（）”将RGB图像转换为灰度

2. 通过应用模糊函数“ GaussianBlur（）”来消除灰度图像中的噪声

3. 最后将“ Canny（）”函数应用于模糊图像以获得边缘

边缘检测过程的输出如下所示：

请注意，尽管已识别出脑图片段，但仍有许多不需要的边缘需要消除，并且某些边缘之间有间隙需要封闭。

解决这个问题的一种常用方法是形态转换，它涉及在图像上使用一系列的扩张和腐蚀来去除不需要的边缘和闭合间隙。

我们在多次迭代中使用OpenCV函数“ dilate（）”和“ erode（）”来获得如下输出。

如我们看到的那样，边缘现在已经完成并且比以前光滑得多。

现在，我们可以使用OpenCV函数“ findContours（）”提取该图像中的轮廓，并仅选择具有以下属性的轮廓：

1. 几何形状是圆形或椭圆形

2. 面积大于某个阈值（在此示例中，值7000可以正常工作）。

对于第一部分，我们将使用OpenCV的“ boundingRect（）”检测每个轮廓的边界矩形，并检查纵横比（高宽比）是否接近1。

现在我们的任务已经完成，但还需要进行一些微调。

通常情况是在一个片段上检测到多个重叠的轮廓，而我们只对一个感兴趣。

使用非极大抑制可以解决此问题，即我们查看所有重叠的轮廓，然后选择面积最大的轮廓作为最终候选轮廓。逻辑非常简单，因此我们不需要任何内置的OpenCV或Python函数。

另一个重要的逻辑是分别识别四个部分，即左上，右上，左下和右下。

这也非常简单，涉及识别图像中心坐标以及每个检测到的片段的质心。对段轮廓进行质心检测需要在轮廓上应用OpenCV “ moments（）”函数，然后使用以下公式计算中心 X，Y坐标：
center_x，center_y =（int（M [“ m10”] / M [” m00”]），int（M [“ m01”] / M [“ m00”]））

将线段质心坐标与图像中心坐标进行比较，可以将四个线段分别放置在各自的位置。

现在我们已经确定了四个部分，我们需要构建图像蒙版，这将使我们能够从原始图像中提取所需的特征。

我们将使用OpenCV函数“ drawContours（）”，将颜色用作白色（R，G，B = 255,2555,255），将厚度用作FILLED（-1）在黑色背景上绘制所有四个线段轮廓。结果如下所示：