基于OpenCV的图像分割处理!

图像阈值化分割是一种传统的最经常使用的图像分割方法,因其实现简单、计算量小、性能较稳定而成为图像分割中最基本和应用最普遍的分割技术。它特别适用于目标和背景占据不一样灰度级范围的图像。它不只能够极大的压缩数据量,并且也大大简化了分析和处理步骤,所以在不少状况下,是进行图像分析、特征提取与模式识别以前的必要的图像预处理过程。python

阈值处理是指剔除图像内像素值高于阈值或者低于阈值得像素点。例如,设定阈值为127,将图像内全部像素值大于127的像素点的值设为255;将图像内全部像素值小于127的像素点的值设为0。ios

图像阈值化的目的是要按照灰度级,对像素集合进行一个划分,获得的每一个子集造成一个与现实景物相对应的区域,各个区域内部具备一致的属性,而相邻区域不具备这种一致属性。这样的划分能够经过从灰度级出发选取一个或多个阈值来实现。c++

学习目标

  • 了解阈值分割基本概念算法

  • 理解最大类间方差法(大津法)、自适应阈值分割的原理app

  • 掌握OpenCV框架下上述阈值分割算法API的使用框架

算法理论介绍

阈值处理

threshold函数

OpenCV使用threshold函数实现阈值化处理。ide


double cv::threshold ( InputArray src,                      OutputArray dst,                      double thresh,                      double maxval,                      int type )

参数函数

  • src — 原图像,8或32位浮点类型的Mat。性能

  • dst —  输出图像,与原始图像具备相同大小和类型。学习

  • thresh —  要设定的阈值

  • maxval   — 当type为THRESH_BINARY或者THRESH_BINARY_INV时,设定的最大值

  • type — 阈值分割的类型

    • THRESH_BINARY;二值化阈值处理:灰度值大于阈值的点,将其灰度值设定为最大值,灰度值小于或等于阈值的点,将其灰度值设定为0

    • THRESH_BINARY_INV;反二值化阈值处理:灰度值大于阈值的点,将其灰度值设定为0,灰度值小于或等于阈值的点,将其灰度值设定为最大值

    • THRESH_TRUNC;截断阈值化处理:灰度值大于阈值的点,将其灰度值设定为阈值,灰度值小于或等于阈值的点,其灰度值保持不变

    • THRESH_TOZERO;低阈值零处理:灰度值大于阈值的点,其灰度值保持不变,灰度值小于或等于阈值的点,将其灰度值设定为0

    • THRESH_TOZERO_INV;高阈值零处理:灰度值大于阈值的点,将其灰度值设定为0,灰度值小于或等于阈值的点,其灰度值保持不变

以下表:

OTSU(大津法)

使用threshold进行阈值处理时,须要自定义一个阈值,并以此阈值做为图像阈值处理的依据 。一般状况下对于色彩均衡的图像,直接将阈值设为127便可,但有时图像灰度级的分布是不均衡的,若是此时还将阈值设为127,那么阈值处理的结果就是失败的。因此须要找出图像的最佳的分割阈值。OTSU就是得到最佳阈值的方法。OTSU(大津法)是一种肯定图像二值化分割阈值的算法,由日本学者大津于1979年提出。从大津法的原理上来说,该方法又称做最大类间方差法,由于按照大津法求得的阈值进行图像二值化分割后,前景与背景图像的类间方差最大。它被认为是图像分割中阈值选取的最佳算法,计算简单,不受图像亮度和对比度的影响,所以在数字图像处理上获得了普遍的应用。它是按图像的灰度特性,将图像分红背景和前景两部分。因方差是灰度分布均匀性的一种度量,背景和前景之间的类间方差越大,说明构成图像的两部分的差异越大,当部分前景错分为背景或部分背景错分为前景都会致使两部分差异变小。所以,使类间方差最大的分割意味着错分几率最小。OTSU 是求图像全局阈值的最佳方法,适用于大部分须要求图像全局阈值的场合。缺点:对图像噪声敏感;只能针对单一目标分割;当图像中的目标与背景的面积相差很大时,表现为直方图没有明显的双峰,或者两个峰的大小相差很大,分割效果不佳,或者目标与背景的灰度有较大的重叠时也不能准确的将目标与背景分开。致使这种现象出现的缘由是该方法忽略了图像的空间信息,同时该方法将图像的灰度分布做为分割图像的依据,于是对噪声也至关敏感。因此,在实际应用中,老是将其与其余方法结合起来使用。

图像直方图

效果:

图像直方图:

效果:


OTSU求阈值过程:

假设图像的像素个数为M×N。假设存在阈值T将图像全部像素分为两类C1(像素值小于T)和C2(像素值大于T)。假设图片背景较暗,则C1类为背景,C2类为前景。像素被分为C1和C2类的几率分别为p一、p2。图像中属于C1类的像素个数记做N1,其平均灰度;属于C2类的的像素个数记做N2,其平均灰度为。图像的总平均灰度记为,类间方差记为所以有以下关系式:

带入类间方差公式,化简,能够获得:

L为灰度级数,为灰度级为的像素点数

小于或等于灰度级K的累加均值为:

因此,

类间方差公式能够化为:

图片

得使最大的灰度级K,就是OTSU的阈值。OTSU方法会遍历全部灰度级,找到最佳阈值

自适应阈值处理

前面介绍了OTSU算法,但这算法还属于全局阈值法,即整张图片只有一个阈值。因此对于某些光照不均的图像,这种方法没法获得清晰有效的阈值分割结果图像,以下图:

图片

显然,这样的阈值处理结果不是咱们想要的,因此须要使用变化的阈值对图像进行分割,这种技术称为自适应阈值处理方式。它的思想不是计算全局图像的阈值,而是根据图像不一样区域亮度分布,计算其局部阈值,因此对于图像不一样区域,可以自适应计算不一样的阈值,所以被称为自适应阈值法。

肯定局部阈值的方法:计算每一个像素点周围临近区域的加权平均值得到阈值,并使用该阈值对该像素点进行处理。

adaptiveThreshold函数

OpenCV提供了adaptiveThreshold函数实现自适应阈值处理。


void adaptiveThreshold(InputArray src, OutputArray dst,                      double maxValue,                      int adaptiveMethod,                      int thresholdType,                      int blockSize, double C)

参数:

  • src — 原图像,8或32位浮点类型的Mat。必须为单通道灰度图像。

  • dst — 输出图像,与原始图像具备相同大小和类型。

  • maxValue — 像素最大值

  • adaptiveMethod — 自适应方法,只有两种:THRESH_BINARY 和THRESH_BINARY_INV。

  • thresholdType — 阈值计算方式,有两种:

    • ADAPTIVE_THRESH_MEAN_C:计算方法是计算出领域内的像素平均值再减去C的值

    • ADAPTIVE_THRESH_GAUSSIAN_C:计算方法是计算出领域内像素的高斯均值再减去C的值

  • blockSize — 表示一个像素在计算阈值时使用的邻域尺寸,一般为三、五、7。

  • C — 常数,用均值或高斯计算阈值后,再减去C就是最终阈值。

基于OpenCV的实现

c++实现

1. 阈值处理


#include <opencv2/opencv.hpp>#include <iostream>using namespace cv;using namespace std;int main(){//载入图像Mat img = imread("D:\\yt\\picture\\threshold\\s.jpg");if (img.empty()){cout << "Error: Could not load image" << endl;return 0;}Mat gray;cvtColor(img, gray, COLOR_BGR2GRAY);//先转为灰度图Mat dst1,dst2,dst3,dst4,dst5;threshold(gray, dst1, 127, 255, THRESH_BINARY);//二值化阈值处理threshold(gray, dst2, 127, 255, THRESH_BINARY_INV);//反二值化阈值处理threshold(gray, dst3, 127, 255, THRESH_TRUNC);//截断阈值化处理threshold(gray, dst4, 127, 255, THRESH_TOZERO_INV);//超阈值零处理threshold(gray, dst5, 127, 255, THRESH_TOZERO);//低阈值零处理//显示图像imshow("gray_image", gray);imshow("THRESH_BINARY", dst1);imshow("THRESH_BINARY_INV", dst2);imshow("THRESH_TRUNC", dst3);imshow("THRESH_TOZERO_INV", dst4);imshow("THRESH_TOZERO", dst5);waitKey(0);return 0;}

效果

二值化阈值处理:

反二值化阈值处理:

截断阈值化处理:

超阈值零处理:

低阈值零处理:

2. OTSU处理

在OpenCV中,设定参数type为“THRESH_OTSU”便可实现OTSU方式的阈值分割。且设定阈值thresh为0。


#include <opencv2/opencv.hpp>#include <iostream>using namespace cv;using namespace std;int main(){//载入图像Mat img = imread("D:\\yt\\picture\\threshold\\s.jpg");if (img.empty()){cout << "Error: Could not load image" << endl;return 0;}Mat gray;cvtColor(img, gray, COLOR_BGR2GRAY);//先转为灰度图Mat dst;threshold(gray, dst, 0, 255, THRESH_OTSU);//OTSU
//显示图像imshow("gray_image", gray);imshow("THRESH_OTSU", dst);waitKey(0);return 0;}

效果

3. 自适应阈值处理


#include <opencv2/opencv.hpp>#include <iostream>using namespace cv;using namespace std;int main(){//载入图像Mat img = imread("D:\\yt\\picture\\threshold\\1.jpg");if (img.empty()){cout << "Error: Could not load image" << endl;return 0;}Mat gray;cvtColor(img, gray, COLOR_BGR2GRAY);//先转为灰度图Mat dst;adaptiveThreshold(gray, dst, 255, ADAPTIVE_THRESH_MEAN_C, THRESH_BINARY, 7, 10);//建立窗口,WINDOW_NORMAL使窗口能够自由调节大小namedWindow("gray_image",WINDOW_NORMAL);namedWindow("adaptiveThreshold", WINDOW_NORMAL);//显示图像imshow("gray_image", gray);imshow("adaptiveThreshold", dst);
waitKey(0);return 0;}

效果

进阶实现(根据原理本身实现)

实现示例(c++)

1. OTSU处理


#include <iostream>#include <opencv2/core.hpp>#include <opencv2/highgui.hpp>#include <opencv2/imgproc.hpp>
int Otsu(cv::Mat& src, cv::Mat& dst, int thresh){const int Grayscale = 256;int graynum[Grayscale] = { 0 };int r = src.rows;int c = src.cols;for (int i = 0; i < r; ++i){const uchar* ptr = src.ptr<uchar>(i);for (int j = 0; j < c; ++j){        //直方图统计graynum[ptr[j]]++;}}
  double P[Grayscale] = { 0 };  double PK[Grayscale] = { 0 };double MK[Grayscale] = { 0 };double srcpixnum = r*c, sumtmpPK = 0, sumtmpMK = 0;for (int i = 0; i < Grayscale; ++i){P[i] = graynum[i] / srcpixnum;   //每一个灰度级出现的几率PK[i] = sumtmpPK + P[i];         //几率累计和sumtmpPK = PK[i];MK[i] = sumtmpMK + i*P[i];       //灰度级的累加均值                                                                                                                                                                                                                                                                                                                                                                                                        sumtmpMK = MK[i];}
//计算类间方差double Var=0;for (int k = 0; k < Grayscale; ++k){if ((MK[Grayscale-1] * PK[k] - MK[k])*(MK[Grayscale-1] * PK[k] - MK[k]) / (PK[k] * (1 - PK[k])) > Var){Var = (MK[Grayscale-1] * PK[k] - MK[k])*(MK[Grayscale-1] * PK[k] - MK[k]) / (PK[k] * (1 - PK[k]));thresh = k;}}
//阈值处理src.copyTo(dst);for (int i = 0; i < r; ++i){    uchar* ptr = dst.ptr<uchar>(i);for (int j = 0; j < c; ++j){if (ptr[j]> thresh)ptr[j] = 255;elseptr[j] = 0;}}return thresh;}

int main(){cv::Mat src = cv::imread("D:\\yt\\picture\\threshold\\1.jpg");if (src.empty()){return -1;}if (src.channels() > 1)cv::cvtColor(src, src, CV_RGB2GRAY);
cv::Mat dst;int thresh=0;double t2 = (double)cv::getTickCount();thresh=Otsu(src , dst, thresh); //Otsu
cv::namedWindow("src", CV_WINDOW_NORMAL);cv::imshow("src", src);cv::namedWindow("dst", CV_WINDOW_NORMAL);cv::imshow("dst", dst);cv::waitKey(0);}

2. 自适应阈值处理


#include <iostream>#include <opencv2/core.hpp>#include <opencv2/highgui.hpp>#include <opencv2/imgproc.hpp>
enum adaptiveMethod{meanFilter,gaaussianFilter,medianFilter};
void AdaptiveThreshold(cv::Mat& src, cv::Mat& dst, double Maxval, int Subsize, double c, adaptiveMethod method = meanFilter){
if (src.channels() > 1)cv::cvtColor(src, src, CV_RGB2GRAY);
cv::Mat smooth;switch (method){case  meanFilter:cv::blur(src, smooth, cv::Size(Subsize, Subsize));  //均值滤波break;case gaaussianFilter:cv::GaussianBlur(src, smooth, cv::Size(Subsize, Subsize),0,0); //高斯滤波break;case medianFilter:cv::medianBlur(src, smooth, Subsize);   //中值滤波break;default:break;}
smooth = smooth - c;
//阈值处理src.copyTo(dst);for (int r = 0; r < src.rows;++r){const uchar* srcptr = src.ptr<uchar>(r);const uchar* smoothptr = smooth.ptr<uchar>(r);uchar* dstptr = dst.ptr<uchar>(r);for (int c = 0; c < src.cols; ++c){if (srcptr[c]>smoothptr[c]){dstptr[c] = Maxval;}elsedstptr[c] = 0;}}
}
int main(){cv::Mat src = cv::imread("D:\\yt\\picture\\threshold\\1.jpg");if (src.empty()){return -1;}if (src.channels() > 1)cv::cvtColor(src, src, CV_RGB2GRAY);
cv::Mat dst;AdaptiveThreshold(src, dst, 255, 21, 10, meanFilter);
cv::namedWindow("src", CV_WINDOW_NORMAL);cv::imshow("src", src);cv::namedWindow("dst", CV_WINDOW_NORMAL);cv::imshow("dst", dst);cv::waitKey(0);}

python实现

与c++不一样,python中函数cv2.threshold的返回值有两个


retval,dst = cv2.threshold(src,thresh,maxval,type)
  • retval — 返回的阈值

  • dst — 阈值处理的输出图像

1. 二值化阈值处理


import cv2if __name__ == "__main__":   img = cv2.imread('D:/yt/picture/threshold/lena.bmp')   t,dst1 = cv2.threshold(img,127,255,cv2.THRESH_BINARY)   # 显示图像   cv2.imshow("origin image", img)   cv2.imshow("THRESH_BINARY", dst1)
  # 保存图像   cv2.imwrite("D:/yt/picture/threshold/THRESH_BINARY.bmp", dst1)   cv2.waitKey(0)   cv2.destroyAllWindows()

效果

2. 反二值化阈值处理


import cv2if __name__ == "__main__":   img = cv2.imread('D:/yt/picture/threshold/lena.bmp')   t,dst1 = cv2.threshold(img,127,255,cv2.THRESH_BINARY_INV)    # 显示图像   cv2.imshow("origin image", img)   cv2.imshow("THRESH_BINARY_INV", dst1)    # 保存图像   cv2.imwrite("D:/yt/picture/threshold/THRESH_BINARY_INV.bmp", dst1)   cv2.waitKey(0)   cv2.destroyAllWindows()

效果

3. 截断阈值化处理


import cv2if __name__ == "__main__":   img = cv2.imread('D:/yt/picture/threshold/lena.bmp')   t,dst1 = cv2.threshold(img,127,255,cv2.THRESH_TRUNC)
  # 显示图像   cv2.imshow("origin image", img)   cv2.imshow("THRESH_TRUNC", dst1)
  # 保存图像   cv2.imwrite("D:/yt/picture/threshold/THRESH_TRUNC.bmp", dst1)   cv2.waitKey(0)   cv2.destroyAllWindows()

效果

4. 超阈值零处理


import cv2if __name__ == "__main__":   img = cv2.imread('D:/yt/picture/threshold/lena.bmp')   t,dst1 = cv2.threshold(img,127,255,cv2.THRESH_TOZERO_INV)
  # 显示图像   cv2.imshow("origin image", img)   cv2.imshow("THRESH_TOZERO_INV", dst1)
  # 保存图像   cv2.imwrite("D:/yt/picture/threshold/THRESH_TOZERO_INV.bmp", dst1)   cv2.waitKey(0)   cv2.destroyAllWindows()

效果

5. 低阈值零处理


import cv2if __name__ == "__main__":   img = cv2.imread('D:/yt/picture/threshold/lena.bmp')   t,dst1 = cv2.threshold(img,127,255,cv2.THRESH_TOZERO)
  # 显示图像   cv2.imshow("origin image", img)   cv2.imshow("THRESH_TOZERO", dst1)
  # 保存图像   cv2.imwrite("D:/yt/picture/threshold/THRESH_TOZERO.bmp", dst1)   cv2.waitKey(0)   cv2.destroyAllWindows()

效果

6. OTSU处理

在OpenCV中,给参数type多传递一个参数“THRESH_OTSU”便可实现OTSU方式的阈值分割。且设定阈值thresh为0。


import cv2import numpy as npif __name__ == "__main__":   img = cv2.imread('D:/yt/picture/threshold/tiffany.bmp')   img = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)#原图像不是灰度图,必须先转换为灰度图   #普通二值化阈值处理   t1, dst1 = cv2.threshold(img,127,255,cv2.THRESH_BINARY)   #采用OTSU的处理   t2, dst2 = cv2.threshold(img, 0, 255, cv2.THRESH_BINARY+cv2.THRESH_OTSU)
  # 建立窗口   cv2.namedWindow("origin image",cv2.WINDOW_NORMAL)#cv2.WINDOW_NORMAL使窗口大小可调整   cv2.namedWindow("THRESH_TOZERO",cv2.WINDOW_NORMAL)   cv2.namedWindow("THRESH_OTSU",cv2.WINDOW_NORMAL)   # 显示图像   cv2.imshow("origin image", img)   cv2.imshow("THRESH_TOZERO", dst1)   cv2.imshow("THRESH_OTSU",dst2)
  cv2.waitKey(0)   cv2.destroyAllWindows()

效果:

7. 自适应阈值处理


import cv2import numpy as npif __name__ == "__main__":   img = cv2.imread('D:/yt/picture/threshold/computer.jpg')   img = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)#原图像不是灰度图,必须先转换为灰度图   #普通二值化阈值处理   t1, dst1 = cv2.threshold(img,127,255,cv2.THRESH_BINARY)   #自适应阈值处理,采用均值计算阈值   dst2 = cv2.adaptiveThreshold(img,255,cv2.ADAPTIVE_THRESH_MEAN_C,cv2.THRESH_BINARY,5,3)   #自适应阈值处理,采用高斯均值计算阈值   dst3 = cv2.adaptiveThreshold(img,255,cv2.ADAPTIVE_THRESH_GAUSSIAN_C,cv2.THRESH_BINARY,5,3)
  # 建立窗口   cv2.namedWindow("origin image",cv2.WINDOW_NORMAL)#cv2.WINDOW_NORMAL使窗口大小可调整   cv2.namedWindow("THRESH_BINARY",cv2.WINDOW_NORMAL)   cv2.namedWindow("MEAN_C",cv2.WINDOW_NORMAL)   cv2.namedWindow("GAUSSIAN_C", cv2.WINDOW_NORMAL)   # 显示图像   cv2.imshow("origin image", img)   cv2.imshow("THRESH_BINARY", dst1)   cv2.imshow("MEAN_C",dst2)   cv2.imshow("GAUSSIAN_C", dst3)
  cv2.waitKey(0)   cv2.destroyAllWindows()

效果

图片

相关文章
相关标签/搜索