基于Opencv實現Matlab的bwmorph中的bridge操作

2019-11-14 10:11:42

字體：大中小

來源：轉載

供稿：網友

工程需要，但是不能用matlab，所以用Opencv實現了一下bridge，也不難。

一. bridge操作的定義

在官方文檔https://cn.mathworks.com/help/images/ref/bwmorph.html中，bridge操作的定義如下： Bridges unconnected pixels, that is, sets 0-valued pixels to 1 if they have two nonzero neighbors that are not connected. For example: 1 0 0 1 0 1 0 0 1 becomes 1 1 0 1 1 1 0 1 1 也就是說，在一個二值化后的圖像中，對于任意一個值為0的像素點，如果它周圍(4連通或8連通)存在至少2個值為1且屬于不同連通區域的像素點，則把這個像素點值修改為1。

二. 實現思路

按照定義把所有值為0的像素點都走一遍就ok了，注意每次修改后，需要更新連通區域的信息。

三. 代碼實現

#include <iostream>#include <opencv2/core/core.hpp>#include <opencv2/highgui/highgui.hpp>#include <opencv2/imgPRoc/imgproc.hpp>#include <set>using namespace cv;using namespace std;const int dx[] = { 0, 1, 0, -1, 1, -1, 1, -1 };const int dy[] = { 1, 0, -1, 0, 1, -1, -1, 1 };set<int> get_neighbors1(Mat src, Mat label_mat, int x, int y, int connectivity = 8) { set<int> neighbors; for (int i = 0; i < connectivity; ++i) { int tmp_x = x + dx[i]; int tmp_y = y + dy[i]; if (tmp_x >= 0 && tmp_x < src.rows && tmp_y >= 0 && tmp_y < src.cols && src.at<uchar>(tmp_x, tmp_y) == 255) { neighbors.insert(label_mat.at<int>(tmp_x, tmp_y)); } } return neighbors;}void bridge(Mat src, Mat &dst, int connectivity = 8) { Mat label_mat; connectedComponents(src, label_mat, connectivity); dst = src.clone(); for (int i = 0; i < src.rows; ++i) { for (int j = 0; j < src.cols; ++j) { uchar pixel = src.at<uchar>(i, j); if (pixel == 0) { if (get_neighbors1(src, label_mat, i, j, 4).size() >= 2) { dst.at<uchar>(i, j) = 255; connectedComponents(dst, label_mat, connectivity); } } } }}int main() { Mat image = imread("C://Users//whai//Desktop//1.png"); cvtColor(image, image, COLOR_BGR2GRAY); // Canny(image, image, 100, 200, 3); imshow("修復前", image); waitKey(); Mat dst; bridge(image, dst, 4); imshow("修復后", dst); imwrite("C://Users//whai//Desktop//2.png", dst); waitKey(); return 0;}