基于距离变换和分水岭算法的图像分割

最新推荐文章于 2024-04-28 22:27:23 发布

zzyczzyc

最新推荐文章于 2024-04-28 22:27:23 发布

阅读量812

点赞数

分类专栏： opencv

本文链接：https://blog.csdn.net/zzyczzyc/article/details/86716997

版权

本文介绍了基于距离变换和分水岭算法的图像分割过程。首先，将背景白色部分转换为黑色，然后对图像进行锐化和二值化。接着，通过距离变换和阈值处理提取前景对象。利用marker为分水岭算法做准备，最后应用OpenCV的distanceTransform和watershed函数进行图像分割。

摘要由CSDN通过智能技术生成

原理

代码

#include <opencv2/opencv.hpp>
#include <iostream>


using namespace cv;
using namespace std;

double maxVal, minVal;

int main(void)
{
   
	
    Mat src = imread("../res/cards.png");
    if(src.empty())
    {
   
	cout << "can't load the image" << endl;
	return -1;
    }

    imshow("Source Image", src);
    
    for(int i=0; i<src.rows; i++)
    {
   
	for(int j=0; j<src.cols; j++)
	{
   
	    if(src.at<Vec3b>(i,j) == Vec3b(255,255,255))
	    {
   
		src.at<Vec3b>(i,j) = Vec3b(0,0,0);
	    }
	}
    }
    imshow("Black Background Image", src);
	

    Mat kernel = (Mat_<float>(3,3) << 1, 1, 1,
				      1,-8, 1,
				      1, 1, 1);

    Mat imgLaplacian;
    filter2D(src, imgLaplacian, CV_32F, kernel);
    Mat sharp;
    src.convertTo(sharp, CV_32F);
    Mat imgResult = sharp - imgLaplacian;

    imgResult.convertTo(imgResult, CV_8UC3);
    imgLaplacian.convertTo(imgLaplacian, CV_8UC3);
    imshow("New Sharped Image", imgResult);


    Mat bw;
    cvtColor(imgResult, bw, COLOR_BGR2GRAY);
    threshold(bw, bw, 40, 255,  THRESH_BINARY|THRESH_OTSU);
    imshow("Binary Image", bw);

    Mat dist;
    cv::distanceTransform(bw, dist, DIST_L2, 3);
    normalize(dist, dist, 0, 1.0, NORM_MINMAX);
    imshow("Distance