實踐教程｜基于OpenCV提取特定區(qū)域方法匯總

今天我們將一起探究如何使用OpenCV和Python從圖像中提取感興趣區(qū)域（ROI）。

在之間的文章中，我們完成了圖像邊緣提取，例如從臺球桌中提取桌邊。使用了簡單的OpenCV函數(shù)即可完成這項任務(wù)，例如inRange、findContours、boundingRect、minAreaRect、minEnclosingCircle、circle、HoughLines、line等，都可以。

今天我們的任務(wù)是從包含患者大腦活動快照的圖像中提取所需的片段。之后可以將該提取的過程應(yīng)用于其他程序中，例如診斷健康與否的機(jī)器學(xué)習(xí)模型。

因此，讓我們從查看輸入圖像開始。這是由神經(jīng)科學(xué)領(lǐng)域的醫(yī)療儀器生成的典型報告，該儀器使用傳感器檢測來自患者大腦的信號并將其顯示為彩色地圖。通常，有四張圖片，所有圖片都描繪了某個特征并一起分析以進(jìn)行診斷。

從上面的圖像中，我們只想提取與四個地圖（頭部掃描）相對應(yīng)的區(qū)域，而將其他所有內(nèi)容都排除在外。因此，讓我們開始吧。

第一步是檢測我們要提取的片段的邊緣。這是一個多步驟過程，如下所述：

1. 使用“ cvtColor（）”將RGB圖像轉(zhuǎn)換為灰度

2. 通過應(yīng)用模糊函數(shù)“ GaussianBlur（）”來消除灰度圖像中的噪聲

3. 最后將“ Canny（）”函數(shù)應(yīng)用于模糊圖像以獲得邊緣

邊緣檢測過程的輸出如下所示：

請注意，盡管已識別出腦圖片段，但仍有許多不需要的邊緣需要消除，并且某些邊緣之間有間隙需要封閉。

解決這個問題的一種常用方法是形態(tài)轉(zhuǎn)換，它涉及在圖像上使用一系列的擴(kuò)張和腐蝕來去除不需要的邊緣和閉合間隙。

我們在多次迭代中使用OpenCV函數(shù)“ dilate（）”和“ erode（）”來獲得如下輸出。

如我們看到的那樣，邊緣現(xiàn)在已經(jīng)完成并且比以前光滑得多。

現(xiàn)在，我們可以使用OpenCV函數(shù)“ findContours（）”提取該圖像中的輪廓，并僅選擇具有以下屬性的輪廓：

1. 幾何形狀是圓形或橢圓形

2. 面積大于某個閾值（在此示例中，值7000可以正常工作）。

對于第一部分，我們將使用OpenCV的“ boundingRect（）”檢測每個輪廓的邊界矩形，并檢查縱橫比（高寬比）是否接近1。

現(xiàn)在我們的任務(wù)已經(jīng)完成，但還需要進(jìn)行一些微調(diào)。

通常情況是在一個片段上檢測到多個重疊的輪廓，而我們只對一個感興趣。

使用非極大抑制可以解決此問題，即我們查看所有重疊的輪廓，然后選擇面積最大的輪廓作為最終候選輪廓。邏輯非常簡單，因此我們不需要任何內(nèi)置的OpenCV或Python函數(shù)。

另一個重要的邏輯是分別識別四個部分，即左上，右上，左下和右下。

這也非常簡單，涉及識別圖像中心坐標(biāo)以及每個檢測到的片段的質(zhì)心。對段輪廓進(jìn)行質(zhì)心檢測需要在輪廓上應(yīng)用OpenCV “ moments（）”函數(shù)，然后使用以下公式計算中心 X，Y坐標(biāo)：
center_x，center_y =（int（M [“ m10”] / M [” m00”]），int（M [“ m01”] / M [“ m00”]））

將線段質(zhì)心坐標(biāo)與圖像中心坐標(biāo)進(jìn)行比較，可以將四個線段分別放置在各自的位置。

現(xiàn)在我們已經(jīng)確定了四個部分，我們需要構(gòu)建圖像蒙版，這將使我們能夠從原始圖像中提取所需的特征。

我們將使用OpenCV函數(shù)“ drawContours（）”，將顏色用作白色（R，G，B = 255,2555,255），將厚度用作FILLED（-1）在黑色背景上繪制所有四個線段輪廓。結(jié)果如下所示：

在原始圖像上應(yīng)用此蒙版可以在我們選擇的背景（例如黑色或白色）上為我們提供所需的分段。

對于黑色背景，我們創(chuàng)建一個黑色畫布，然后使用OpenCV函數(shù)“ bitwise_and（）”以及先前獲得的蒙版在其上進(jìn)行繪制。

對于白色背景，我們首先創(chuàng)建一個白色畫布，然后通過使用OpenCV函數(shù)“ drawContours（）”繪制輪廓為黑色（R，G，B = 0,0,0）且厚度為FILLED的輪廓，如下所示創(chuàng)建顏色反轉(zhuǎn)的蒙版（-1）。

然后，我們使用OpenCV “ add（）”函數(shù)將此反向蒙版添加到先前獲得的黑色背景中，并獲得相同的結(jié)果，但使用白色背景。

到此為止，我們總結(jié)了幾種方法，可以輕松地從圖像中提取感興趣區(qū)域。應(yīng)當(dāng)注意，在具有變化的復(fù)雜度的其他圖像的情況下，上面使用的方法可以進(jìn)行修改。

如果覺得有用，就請分享到朋友圈吧！