• C++手敲Roberts_Prewitt_Sobel实现阈值分割


    使用OPENCV,编写代码,学习图像二值化算法,以及边缘检测算法,进行图像的分割。

    下面主要介绍Robert算子的实现过程:

    ①任务分析

    调入并显示图像;使用Roberts 算子对图像进行边缘检测处理; Roberts 算子为一对模板:

    c35d8dfb3e764d80981995b7803370d9.png

     相应的矩阵为:rh = [0 1;-1 0]; rv = [1 0;0 -1];这里的rh 为水平Roberts 算子,rv为垂直Roberts 算子。分别显示处理后的水平边界和垂直边界检测结果;用“欧几里德距离”和“街区距离”方式计算梯度的模,并显示检测结果;对于检测结果进行二值化处理,并显示处理结果。

    ②代码实现

    先加载需要的库

    1. #include
    2. #include
    3. using namespace cv;

    接着定义一个基于Roberts算子实现阈值分割的函数,其功能包括显示用街区距离和欧几里得距离得出的梯度图、选择一个合适的阈值实现图像分割。
    其中街区距离就是水平梯度和垂直梯度的和:

    Robert_L1.at(row, col) = saturate_cast(fabs(img.at(row, col) - img.at(row + 1, col + 1))+ fabs(img.at(row, col + 1) - img.at(row + 1, col)));

    saturate_cast()函数的作用是限制像素值的范围,把大于255和小于0的情况分别赋值255和0,避免了像素值溢出的问题。

    而欧几里得距离就是水平梯度和垂直梯度的平方和再开方:

    Robert_L2.at(row, col) = saturate_cast(sqrt(pow(img.at(row, col) - img.at(row + 1, col + 1),2) + pow(img.at(row, col + 1) - img.at(row + 1, col),2)));

    具体代码实现:

    1. Mat Robert_L1 = Mat::zeros(Size(img.cols, img.rows), img.type()); //用于计算城市距离的空白图像
    2. Mat Robert_L2 = Mat::zeros(Size(img.rows, img.cols), img.type()); //用于计算欧几里得距离的空白图像
    3. for (int row = 0; row < img.rows-1; row++) {
    4. for (int col = 0; col < img.cols-1; col++) { // 由于像素之间的加减可能会小于零,因此记得加上绝对值函数fabs()
    5. Robert_L1.at(row, col) = saturate_cast(fabs(img.at(row, col) - img.at(row + 1, col + 1)) + fabs(img.at(row, col + 1) - img.at(row + 1, col)));
    6. }
    7. }
    8. for (int row = 0; row < img.rows - 1; row++) {
    9. for (int col = 0; col < img.cols - 1; col++) {
    10. Robert_L2.at(row, col) = saturate_cast(sqrt(pow(img.at(row, col) - img.at(row + 1, col + 1),2) + pow(img.at(row, col + 1) - img.at(row + 1, col),2)));
    11. }
    12. }
    13. imshow("Robert图像(街区距离)", Robert_L1);
    14. imshow("Robert图像(欧几里得距离)", Robert_L2);

    结果如下:

    bf4442d47066443fb3205fb0c54c008b.png (原图)
    ca9a9c1fac0e443990fdf5dca253273e.png8937aa6f6d434416881a7551e58c62d5.png

     接下来要选择一个合适的阈值用于图像的二值化。选择阈值的方式有很多暴力遍历(不推荐)、调用函数法(直接输出直方图看每一个像素值的数量,方便主观判断)和数组法。这里介绍一下数组法:创建一个256大小的一维数组arr[256],每一个元素依据索引代表对应的像素值的个数。就比如说假设一共有10000个点像素值为0,那么arr[0] = 10000。先对欧几里得距离实现二值化,实现如下:

    1. int pixel_num[256] = {0}; //数组记得初始化为0,即未统计数量之前各个像素的个数都是0
    2. for (int row = 0; row < img.rows ; row++) {
    3. for (int col = 0; col < img.cols ; col++) {
    4. pixel_num[Robert_L2.at(row, col)] += 1; //遍历到的像素值作为索引,次数+1
    5. }
    6. }
    7. int times = 0; //定义遍历数组的变量,从0开始,依次输出0到255每个像素值的数目
    8. while (times <= 255) {
    9. cout <<"像素值"<"的数目为: " << pixel_num[times] << endl; // 遍历输出
    10. times++; // 不要忘了自增
    11. }

     输出结果:
    766367b54f6a4c0aaf01f857a8a8df1b.png

     可以看出,像素值9和像素值10差异变化较大。因此可以选择像素值10作为分割图像的阈值,比像素值10小的像素全部赋值为0,比像素值大的全部赋值255:

    1. for (int row = 0; row < img.rows; row++) {
    2. for (int col = 0; col < img.cols; col++) { //遍历所有像素点进行判断
    3. if (Robert_L2.at(row, col) < 10) {
    4. Robert_L2.at(row, col) = 0; // 小于阈值赋值为0,否则赋值255
    5. }
    6. else {
    7. Robert_L2.at(row, col) = 255;
    8. }
    9. }
    10. }
    11. imshow("欧几里得阈值分割", Robert_L2);

    345edbb7b0af4949b2b06f6fa1c34b0d.png

     可以看到阈值分割效果不错,但是不足之处也很明显:图像因此留下了很多离散点,这是阈值选择得不够大的原因。于是我又偷偷的把阈值调整为15:
    3771e8982e294de39f4d33b87df4fc5e.png在保证图片不失真的情况下,15作为阈值要明显好于10。

    同理,街区距离的做法和欧几里得距离类似,也是通过数组得到较为可信的阈值之后再细微调整。
    街区距离的实现过程以及结果如下所示:

    1. int pixel_num02[256] = { 0 }; //数组记得初始化为0,即未统计数量之前各个像素的个数都是0
    2. for (int row = 0; row < img.rows; row++) {
    3. for (int col = 0; col < img.cols; col++) {
    4. pixel_num01[Robert_L1.at(row, col)] += 1; //遍历到的像素值作为索引,次数+1
    5. }
    6. }
    7. int time = 0; //定义遍历数组的变量,从0开始,依次输出0到255每个像素值的数目
    8. while (times <= 255) {
    9. cout << "像素值" << time << "的数目为: " << pixel_num02[time] << endl; // 遍历输出
    10. time++; // 不要忘了自增
    11. }
    12. //得到10作为分割阈值
    13. for (int row = 0; row < img.rows; row++) {
    14. for (int col = 0; col < img.cols; col++) { //遍历所有像素点进行判断
    15. if (Robert_L1.at(row, col) < 20) {
    16. Robert_L1.at(row, col) = 0; // 小于阈值赋值为0,否则赋值255
    17. }
    18. else {
    19. Robert_L1.at(row, col) = 255;
    20. }
    21. }
    22. }
    23. imshow("街区距离阈值分割", Robert_L1);


    cf1a2a1d009140b1980a728e08af0f34.png

     注意:阈值的选择是很关键的,如果阈值选择过大的话可能会导致边缘断开的情况,也就是失真。这就需要我们在去噪和失真两个方面做权衡。以下给出了基于Roberts算子的阈值分割函数的完整代码:

    1. void Roberts(Mat& img) { // 基于Roberts算子的阈值分割
    2. Mat Robert_L1 = Mat::zeros(Size(img.cols, img.rows), img.type()); //用于计算城市距离的空白图像
    3. Mat Robert_L2 = Mat::zeros(Size(img.cols, img.rows), img.type()); //用于计算欧几里得距离的空白图像
    4. for (int row = 0; row < img.rows-1; row++) {
    5. for (int col = 0; col < img.cols-1; col++) { // 由于像素之间的加减可能会小于零,因此记得加上绝对值函数fabs()
    6. Robert_L1.at(row, col) = saturate_cast(fabs(img.at(row, col) - img.at(row + 1, col + 1)) + fabs(img.at(row, col + 1) - img.at(row + 1, col)));
    7. }
    8. }
    9. for (int row = 0; row < img.rows - 1; row++) {
    10. for (int col = 0; col < img.cols - 1; col++) {
    11. Robert_L2.at(row, col) = saturate_cast(sqrt(pow(img.at(row, col) - img.at(row + 1, col + 1),2) + pow(img.at(row, col + 1) - img.at(row + 1, col),2)));
    12. }
    13. }
    14. imshow("Robert图像(街区距离)", Robert_L1);
    15. imshow("Robert图像(欧几里得距离)", Robert_L2);
    16. int pixel_num01[256] = {0}; //数组记得初始化为0,即未统计数量之前各个像素的个数都是0
    17. for (int row = 0; row < img.rows ; row++) {
    18. for (int col = 0; col < img.cols ; col++) {
    19. pixel_num01[Robert_L2.at(row, col)] += 1; //遍历到的像素值作为索引,次数+1
    20. }
    21. }
    22. int times = 0; //定义遍历数组的变量,从0开始,依次输出0到255每个像素值的数目
    23. while (times <= 255) {
    24. cout <<"像素值"<"的数目为: " << pixel_num01[times] << endl; // 遍历输出
    25. times++; // 不要忘了自增
    26. }
    27. //得到10作为分割阈值
    28. for (int row = 0; row < img.rows; row++) {
    29. for (int col = 0; col < img.cols; col++) { //遍历所有像素点进行判断
    30. if (Robert_L2.at(row, col) < 15) {
    31. Robert_L2.at(row, col) = 0; // 小于阈值赋值为0,否则赋值255
    32. }
    33. else {
    34. Robert_L2.at(row, col) = 255;
    35. }
    36. }
    37. }
    38. imshow("欧几里得阈值分割", Robert_L2);
    39. int pixel_num02[256] = { 0 }; //数组记得初始化为0,即未统计数量之前各个像素的个数都是0
    40. for (int row = 0; row < img.rows; row++) {
    41. for (int col = 0; col < img.cols; col++) {
    42. pixel_num01[Robert_L1.at(row, col)] += 1; //遍历到的像素值作为索引,次数+1
    43. }
    44. }
    45. int time = 0; //定义遍历数组的变量,从0开始,依次输出0到255每个像素值的数目
    46. while (times <= 255) {
    47. cout << "像素值" << time << "的数目为: " << pixel_num02[time] << endl; // 遍历输出
    48. time++; // 不要忘了自增
    49. }
    50. //得到10作为分割阈值
    51. for (int row = 0; row < img.rows; row++) {
    52. for (int col = 0; col < img.cols; col++) { //遍历所有像素点进行判断
    53. if (Robert_L1.at(row, col) < 20) {
    54. Robert_L1.at(row, col) = 0; // 小于阈值赋值为0,否则赋值255
    55. }
    56. else {
    57. Robert_L1.at(row, col) = 255;
    58. }
    59. }
    60. }
    61. imshow("街区距离阈值分割", Robert_L1);
    62. waitKey(0);
    63. destroyAllWindows();
    64. }

     类似的如Prewitt算子、Sobel算子与上述的Roberts算子类似,其阈值分割的过程都是先用街区距离、欧几里得距离分别算出原始图像的梯度,然后再创建一个一维数组来记录下每个像素值的数目以备接下来的阈值选择操作。最后的阈值选择既需要能有效地对图像进行分割,也要保证图像不会出现失真的现象。

    以下分别是基于Prewitt算子的阈值分割函数、基于Sobel算子的阈值分割函数。

    1. void Prewitt(Mat& img) { // 基于Prewitt算子的阈值分割
    2. Mat Prewitt_L1 = Mat::zeros(Size(img.cols, img.rows), img.type()); //用于计算城市距离的空白图像
    3. Mat Prewitt_L2 = Mat::zeros(Size(img.cols, img.rows), img.type()); //用于计算欧几里得距离的空白图像
    4. for (int row = 1; row < img.rows - 1; row++) {
    5. for (int col = 1; col < img.cols - 1; col++) { // 由于像素之间的加减可能会小于零,因此记得加上绝对值函数fabs()
    6. Prewitt_L1.at(row, col) = saturate_cast(fabs(img.at(row-1, col+1) - img.at(row - 1, col - 1)+ img.at(row , col + 1)- img.at(row, col - 1)+ img.at(row + 1, col + 1)- img.at(row + 1, col - 1)) + fabs(img.at(row+1, col - 1) - img.at(row -1 , col-1)+ img.at(row + 1, col)- img.at(row - 1, col)+ img.at(row + 1, col + 1)- img.at(row - 1, col + 1)));
    7. }
    8. }
    9. for (int row = 1; row < img.rows - 1; row++) {
    10. for (int col = 1; col < img.cols - 1; col++) {
    11. Prewitt_L2.at(row, col) = saturate_cast(sqrt(pow(img.at(row - 1, col + 1) - img.at(row - 1, col - 1) + img.at(row, col + 1) - img.at(row, col - 1)+ img.at(row + 1, col + 1) - img.at(row + 1, col - 1), 2) + pow(img.at(row + 1, col - 1) - img.at(row - 1, col - 1) + img.at(row + 1, col) - img.at(row - 1, col) + img.at(row + 1, col + 1) - img.at(row - 1, col + 1), 2)));
    12. }
    13. }
    14. imshow("Prewitt图像(街区距离)", Prewitt_L1);
    15. imshow("Prewitt图像(欧几里得距离)", Prewitt_L2);
    16. int pixel_num01[256] = { 0 }; //数组记得初始化为0,即未统计数量之前各个像素的个数都是0
    17. for (int row = 0; row < img.rows; row++) {
    18. for (int col = 0; col < img.cols; col++) {
    19. pixel_num01[Prewitt_L2.at(row, col)] += 1; //遍历到的像素值作为索引,次数+1
    20. }
    21. }
    22. int times = 0; //定义遍历数组的变量,从0开始,依次输出0到255每个像素值的数目
    23. while (times <= 255) {
    24. cout << "像素值" << times << "的数目为: " << pixel_num01[times] << endl; // 遍历输出
    25. times++; // 不要忘了自增
    26. }
    27. //得到10作为分割阈值
    28. for (int row = 0; row < img.rows; row++) {
    29. for (int col = 0; col < img.cols; col++) { //遍历所有像素点进行判断
    30. if (Prewitt_Ojld.at(row, col) < 70) {
    31. Prewitt_Ojld.at(row, col) = 0; // 小于阈值赋值为0,否则赋值255
    32. }
    33. else {
    34. Prewitt_L2.at(row, col) = 255;
    35. }
    36. }
    37. }
    38. imshow("欧几里得阈值分割", Prewitt_L2);
    39. int pixel_num02[256] = { 0 }; //数组记得初始化为0,即未统计数量之前各个像素的个数都是0
    40. for (int row = 0; row < img.rows; row++) {
    41. for (int col = 0; col < img.cols; col++) {
    42. pixel_num01[Prewitt_L1.at(row, col)] += 1; //遍历到的像素值作为索引,次数+1
    43. }
    44. }
    45. int time = 0; //定义遍历数组的变量,从0开始,依次输出0到255每个像素值的数目
    46. while (times <= 255) {
    47. cout << "像素值" << time << "的数目为: " << pixel_num02[time] << endl; // 遍历输出
    48. time++; // 不要忘了自增
    49. }
    50. //得到10作为分割阈值
    51. for (int row = 0; row < img.rows; row++) {
    52. for (int col = 0; col < img.cols; col++) { //遍历所有像素点进行判断
    53. if (Prewitt_L1.at(row, col) < 70) {
    54. Prewitt_L1.at(row, col) = 0; // 小于阈值赋值为0,否则赋值255
    55. }
    56. else {
    57. Prewitt_L1.at(row, col) = 255;
    58. }
    59. }
    60. }
    61. imshow("街区距离阈值分割", Prewitt_L1);
    62. waitKey(0);
    63. destroyAllWindows();
    64. }
    1. void Sobel(Mat& img) { // 基于Prewitt算子的阈值分割
    2. Mat Sobel_L1 = Mat::zeros(Size(img.cols, img.rows), img.type()); //用于计算城市距离的空白图像
    3. Mat Sobel_L2 = Mat::zeros(Size(img.cols, img.rows), img.type()); //用于计算欧几里得距离的空白图像
    4. for (int row = 1; row < img.rows - 1; row++) {
    5. for (int col = 1; col < img.cols - 1; col++) { // 由于像素之间的加减可能会小于零,因此记得加上绝对值函数fabs()
    6. Sobel_L1.at(row, col) = saturate_cast(fabs(img.at(row - 1, col + 1) - img.at(row - 1, col - 1) + 2*img.at(row, col + 1) - 2*img.at(row, col - 1) + img.at(row + 1, col + 1) - img.at(row + 1, col - 1)) + fabs(img.at(row + 1, col - 1) - img.at(row - 1, col - 1) + 2*img.at(row + 1, col) - 2*img.at(row - 1, col) + img.at(row + 1, col + 1) - img.at(row - 1, col + 1)));
    7. }
    8. }
    9. for (int row = 1; row < img.rows - 1; row++) {
    10. for (int col = 1; col < img.cols - 1; col++) {
    11. Sobel_L2.at(row, col) = saturate_cast(sqrt(pow(img.at(row - 1, col + 1) - img.at(row - 1, col - 1) + 2*img.at(row, col + 1) - 2*img.at(row, col - 1) + img.at(row + 1, col + 1) - img.at(row + 1, col - 1), 2) + pow(img.at(row + 1, col - 1) - img.at(row - 1, col - 1) + 2*img.at(row + 1, col) - 2*img.at(row - 1, col) + img.at(row + 1, col + 1) - img.at(row - 1, col + 1), 2)));
    12. }
    13. }
    14. imshow("Sobel图像(街区距离)", Sobel_L1);
    15. imshow("Sobel图像(欧几里得距离)", Sobel_L2);
    16. int pixel_num01[256] = { 0 }; //数组记得初始化为0,即未统计数量之前各个像素的个数都是0
    17. for (int row = 0; row < img.rows; row++) {
    18. for (int col = 0; col < img.cols; col++) {
    19. pixel_num01[Sobel_L2.at(row, col)] += 1; //遍历到的像素值作为索引,次数+1
    20. }
    21. }
    22. int times = 0; //定义遍历数组的变量,从0开始,依次输出0到255每个像素值的数目
    23. while (times <= 255) {
    24. cout << "像素值" << times << "的数目为: " << pixel_num01[times] << endl; // 遍历输出
    25. times++; // 不要忘了自增
    26. }
    27. //得到10作为分割阈值
    28. for (int row = 0; row < img.rows; row++) {
    29. for (int col = 0; col < img.cols; col++) { //遍历所有像素点进行判断
    30. if (Sobel_L2.at(row, col) < 70) {
    31. Sobel_L2.at(row, col) = 0; // 小于阈值赋值为0,否则赋值255
    32. }
    33. else {
    34. Sobel_L2.at(row, col) = 255;
    35. }
    36. }
    37. }
    38. imshow("欧几里得阈值分割", Sobel_L2);
    39. int pixel_num02[256] = { 0 }; //数组记得初始化为0,即未统计数量之前各个像素的个数都是0
    40. for (int row = 0; row < img.rows; row++) {
    41. for (int col = 0; col < img.cols; col++) {
    42. pixel_num01[Sobel_L1.at(row, col)] += 1; //遍历到的像素值作为索引,次数+1
    43. }
    44. }
    45. int time = 0; //定义遍历数组的变量,从0开始,依次输出0到255每个像素值的数目
    46. while (times <= 255) {
    47. cout << "像素值" << time << "的数目为: " << pixel_num02[time] << endl; // 遍历输出
    48. time++; // 不要忘了自增
    49. }
    50. //得到10作为分割阈值
    51. for (int row = 0; row < img.rows; row++) {
    52. for (int col = 0; col < img.cols; col++) { //遍历所有像素点进行判断
    53. if (Sobel_L1.at(row, col) < 70) {
    54. Sobel_L1.at(row, col) = 0; // 小于阈值赋值为0,否则赋值255
    55. }
    56. else {
    57. Sobel_L1.at(row, col) = 255;
    58. }
    59. }
    60. }
    61. imshow("街区距离阈值分割", Sobel_L1);
    62. waitKey(0);
    63. destroyAllWindows();
    64. }

    Prewitt算子和Sobel算子大同小异,Sobel是在Prewitt的基础上增加了权重,也就是越靠近当前像素值,那么它所发挥的作用就越大。总体而言的话Prewitt算子和Sobel算子的结果会比Roberts算子的结果要好一些,下面是基于两者的阈值分割结果:
    a53dcba6284848f0b916f72e7f86d7d7.png

    791bb3e6794e4b2db0895b53ebdbc300.png

    2df23ca6a48f431ebeffae46e76e263a.png40f291bad9674e398fb4d4ba3e817d63.png

     总代码如下:

    1. #include
    2. #include
    3. #include
    4. using namespace cv;
    5. using namespace std;
    6. void Roberts(Mat& img);
    7. void Prewitt(Mat& img);
    8. void Sobel(Mat& img);
    9. int main() {
    10. Mat Gray = imread("C:\\Users\\Czhannb\\Desktop\\gray.png", IMREAD_GRAYSCALE);
    11. if (Gray.empty()) {
    12. cout << "读取图片错误!" << endl;
    13. }
    14. else {
    15. imshow("未动工之前:", Gray);
    16. }
    17. //Roberts(Gray);
    18. //Prewitt(Gray);
    19. Sobel(Gray);
    20. return 0;
    21. }
    22. void Roberts(Mat& img) { // 基于Roberts算子的阈值分割
    23. Mat Robert_L1 = Mat::zeros(Size(img.cols, img.rows), img.type()); //用于计算城市距离的空白图像
    24. Mat Robert_L2 = Mat::zeros(Size(img.cols, img.rows), img.type()); //用于计算欧几里得距离的空白图像
    25. for (int row = 0; row < img.rows-1; row++) {
    26. for (int col = 0; col < img.cols-1; col++) { // 由于像素之间的加减可能会小于零,因此记得加上绝对值函数fabs()
    27. Robert_L1.at(row, col) = saturate_cast(fabs(img.at(row, col) - img.at(row + 1, col + 1)) + fabs(img.at(row, col + 1) - img.at(row + 1, col)));
    28. }
    29. }
    30. for (int row = 0; row < img.rows - 1; row++) {
    31. for (int col = 0; col < img.cols - 1; col++) {
    32. Robert_L2.at(row, col) = saturate_cast(sqrt(pow(img.at(row, col) - img.at(row + 1, col + 1),2) + pow(img.at(row, col + 1) - img.at(row + 1, col),2)));
    33. }
    34. }
    35. imshow("Robert图像(街区距离)", Robert_L1);
    36. imshow("Robert图像(欧几里得距离)", Robert_L2);
    37. int pixel_num01[256] = {0}; //数组记得初始化为0,即未统计数量之前各个像素的个数都是0
    38. for (int row = 0; row < img.rows ; row++) {
    39. for (int col = 0; col < img.cols ; col++) {
    40. pixel_num01[Robert_L2.at(row, col)] += 1; //遍历到的像素值作为索引,次数+1
    41. }
    42. }
    43. int times = 0; //定义遍历数组的变量,从0开始,依次输出0到255每个像素值的数目
    44. while (times <= 255) {
    45. cout <<"像素值"<"的数目为: " << pixel_num01[times] << endl; // 遍历输出
    46. times++; // 不要忘了自增
    47. }
    48. //得到10作为分割阈值
    49. for (int row = 0; row < img.rows; row++) {
    50. for (int col = 0; col < img.cols; col++) { //遍历所有像素点进行判断
    51. if (Robert_L2.at(row, col) < 15) {
    52. Robert_L2.at(row, col) = 0; // 小于阈值赋值为0,否则赋值255
    53. }
    54. else {
    55. Robert_L2.at(row, col) = 255;
    56. }
    57. }
    58. }
    59. imshow("欧几里得阈值分割", Robert_L2);
    60. int pixel_num02[256] = { 0 }; //数组记得初始化为0,即未统计数量之前各个像素的个数都是0
    61. for (int row = 0; row < img.rows; row++) {
    62. for (int col = 0; col < img.cols; col++) {
    63. pixel_num01[Robert_L1.at(row, col)] += 1; //遍历到的像素值作为索引,次数+1
    64. }
    65. }
    66. int time = 0; //定义遍历数组的变量,从0开始,依次输出0到255每个像素值的数目
    67. while (times <= 255) {
    68. cout << "像素值" << time << "的数目为: " << pixel_num02[time] << endl; // 遍历输出
    69. time++; // 不要忘了自增
    70. }
    71. //得到10作为分割阈值
    72. for (int row = 0; row < img.rows; row++) {
    73. for (int col = 0; col < img.cols; col++) { //遍历所有像素点进行判断
    74. if (Robert_L1.at(row, col) < 20) {
    75. Robert_L1.at(row, col) = 0; // 小于阈值赋值为0,否则赋值255
    76. }
    77. else {
    78. Robert_L1.at(row, col) = 255;
    79. }
    80. }
    81. }
    82. imshow("街区距离阈值分割", Robert_L1);
    83. waitKey(0);
    84. destroyAllWindows();
    85. }
    86. void Prewitt(Mat& img) { // 基于Prewitt算子的阈值分割
    87. Mat Prewitt_L1 = Mat::zeros(Size(img.cols, img.rows), img.type()); //用于计算城市距离的空白图像
    88. Mat Prewitt_L2 = Mat::zeros(Size(img.cols, img.rows), img.type()); //用于计算欧几里得距离的空白图像
    89. for (int row = 1; row < img.rows - 1; row++) {
    90. for (int col = 1; col < img.cols - 1; col++) { // 由于像素之间的加减可能会小于零,因此记得加上绝对值函数fabs()
    91. Prewitt_L1.at(row, col) = saturate_cast(fabs(img.at(row-1, col+1) - img.at(row - 1, col - 1)+ img.at(row , col + 1)- img.at(row, col - 1)+ img.at(row + 1, col + 1)- img.at(row + 1, col - 1)) + fabs(img.at(row+1, col - 1) - img.at(row -1 , col-1)+ img.at(row + 1, col)- img.at(row - 1, col)+ img.at(row + 1, col + 1)- img.at(row - 1, col + 1)));
    92. }
    93. }
    94. for (int row = 1; row < img.rows - 1; row++) {
    95. for (int col = 1; col < img.cols - 1; col++) {
    96. Prewitt_L2.at(row, col) = saturate_cast(sqrt(pow(img.at(row - 1, col + 1) - img.at(row - 1, col - 1) + img.at(row, col + 1) - img.at(row, col - 1)+ img.at(row + 1, col + 1) - img.at(row + 1, col - 1), 2) + pow(img.at(row + 1, col - 1) - img.at(row - 1, col - 1) + img.at(row + 1, col) - img.at(row - 1, col) + img.at(row + 1, col + 1) - img.at(row - 1, col + 1), 2)));
    97. }
    98. }
    99. imshow("Prewitt图像(街区距离)", Prewitt_L1);
    100. imshow("Prewitt图像(欧几里得距离)", Prewitt_L2);
    101. int pixel_num01[256] = { 0 }; //数组记得初始化为0,即未统计数量之前各个像素的个数都是0
    102. for (int row = 0; row < img.rows; row++) {
    103. for (int col = 0; col < img.cols; col++) {
    104. pixel_num01[Prewitt_L2.at(row, col)] += 1; //遍历到的像素值作为索引,次数+1
    105. }
    106. }
    107. int times = 0; //定义遍历数组的变量,从0开始,依次输出0到255每个像素值的数目
    108. while (times <= 255) {
    109. cout << "像素值" << times << "的数目为: " << pixel_num01[times] << endl; // 遍历输出
    110. times++; // 不要忘了自增
    111. }
    112. //得到10作为分割阈值
    113. for (int row = 0; row < img.rows; row++) {
    114. for (int col = 0; col < img.cols; col++) { //遍历所有像素点进行判断
    115. if (Prewitt_Ojld.at(row, col) < 70) {
    116. Prewitt_Ojld.at(row, col) = 0; // 小于阈值赋值为0,否则赋值255
    117. }
    118. else {
    119. Prewitt_L2.at(row, col) = 255;
    120. }
    121. }
    122. }
    123. imshow("欧几里得阈值分割", Prewitt_L2);
    124. int pixel_num02[256] = { 0 }; //数组记得初始化为0,即未统计数量之前各个像素的个数都是0
    125. for (int row = 0; row < img.rows; row++) {
    126. for (int col = 0; col < img.cols; col++) {
    127. pixel_num01[Prewitt_L1.at(row, col)] += 1; //遍历到的像素值作为索引,次数+1
    128. }
    129. }
    130. int time = 0; //定义遍历数组的变量,从0开始,依次输出0到255每个像素值的数目
    131. while (times <= 255) {
    132. cout << "像素值" << time << "的数目为: " << pixel_num02[time] << endl; // 遍历输出
    133. time++; // 不要忘了自增
    134. }
    135. //得到10作为分割阈值
    136. for (int row = 0; row < img.rows; row++) {
    137. for (int col = 0; col < img.cols; col++) { //遍历所有像素点进行判断
    138. if (Prewitt_L1.at(row, col) < 70) {
    139. Prewitt_L1.at(row, col) = 0; // 小于阈值赋值为0,否则赋值255
    140. }
    141. else {
    142. Prewitt_L1.at(row, col) = 255;
    143. }
    144. }
    145. }
    146. imshow("街区距离阈值分割", Prewitt_L1);
    147. waitKey(0);
    148. destroyAllWindows();
    149. }
    150. void Sobel(Mat& img) { // 基于Prewitt算子的阈值分割
    151. Mat Sobel_L1 = Mat::zeros(Size(img.cols, img.rows), img.type()); //用于计算城市距离的空白图像
    152. Mat Sobel_L2 = Mat::zeros(Size(img.cols, img.rows), img.type()); //用于计算欧几里得距离的空白图像
    153. for (int row = 1; row < img.rows - 1; row++) {
    154. for (int col = 1; col < img.cols - 1; col++) { // 由于像素之间的加减可能会小于零,因此记得加上绝对值函数fabs()
    155. Sobel_L1.at(row, col) = saturate_cast(fabs(img.at(row - 1, col + 1) - img.at(row - 1, col - 1) + 2*img.at(row, col + 1) - 2*img.at(row, col - 1) + img.at(row + 1, col + 1) - img.at(row + 1, col - 1)) + fabs(img.at(row + 1, col - 1) - img.at(row - 1, col - 1) + 2*img.at(row + 1, col) - 2*img.at(row - 1, col) + img.at(row + 1, col + 1) - img.at(row - 1, col + 1)));
    156. }
    157. }
    158. for (int row = 1; row < img.rows - 1; row++) {
    159. for (int col = 1; col < img.cols - 1; col++) {
    160. Sobel_L2.at(row, col) = saturate_cast(sqrt(pow(img.at(row - 1, col + 1) - img.at(row - 1, col - 1) + 2*img.at(row, col + 1) - 2*img.at(row, col - 1) + img.at(row + 1, col + 1) - img.at(row + 1, col - 1), 2) + pow(img.at(row + 1, col - 1) - img.at(row - 1, col - 1) + 2*img.at(row + 1, col) - 2*img.at(row - 1, col) + img.at(row + 1, col + 1) - img.at(row - 1, col + 1), 2)));
    161. }
    162. }
    163. imshow("Sobel图像(街区距离)", Sobel_L1);
    164. imshow("Sobel图像(欧几里得距离)", Sobel_L2);
    165. int pixel_num01[256] = { 0 }; //数组记得初始化为0,即未统计数量之前各个像素的个数都是0
    166. for (int row = 0; row < img.rows; row++) {
    167. for (int col = 0; col < img.cols; col++) {
    168. pixel_num01[Sobel_L2.at(row, col)] += 1; //遍历到的像素值作为索引,次数+1
    169. }
    170. }
    171. int times = 0; //定义遍历数组的变量,从0开始,依次输出0到255每个像素值的数目
    172. while (times <= 255) {
    173. cout << "像素值" << times << "的数目为: " << pixel_num01[times] << endl; // 遍历输出
    174. times++; // 不要忘了自增
    175. }
    176. //得到10作为分割阈值
    177. for (int row = 0; row < img.rows; row++) {
    178. for (int col = 0; col < img.cols; col++) { //遍历所有像素点进行判断
    179. if (Sobel_L2.at(row, col) < 70) {
    180. Sobel_L2.at(row, col) = 0; // 小于阈值赋值为0,否则赋值255
    181. }
    182. else {
    183. Sobel_L2.at(row, col) = 255;
    184. }
    185. }
    186. }
    187. imshow("欧几里得阈值分割", Sobel_L2);
    188. int pixel_num02[256] = { 0 }; //数组记得初始化为0,即未统计数量之前各个像素的个数都是0
    189. for (int row = 0; row < img.rows; row++) {
    190. for (int col = 0; col < img.cols; col++) {
    191. pixel_num01[Sobel_L1.at(row, col)] += 1; //遍历到的像素值作为索引,次数+1
    192. }
    193. }
    194. int time = 0; //定义遍历数组的变量,从0开始,依次输出0到255每个像素值的数目
    195. while (times <= 255) {
    196. cout << "像素值" << time << "的数目为: " << pixel_num02[time] << endl; // 遍历输出
    197. time++; // 不要忘了自增
    198. }
    199. //得到10作为分割阈值
    200. for (int row = 0; row < img.rows; row++) {
    201. for (int col = 0; col < img.cols; col++) { //遍历所有像素点进行判断
    202. if (Sobel_L1.at(row, col) < 70) {
    203. Sobel_L1.at(row, col) = 0; // 小于阈值赋值为0,否则赋值255
    204. }
    205. else {
    206. Sobel_L1.at(row, col) = 255;
    207. }
    208. }
    209. }
    210. imshow("街区距离阈值分割", Sobel_L1);
    211. waitKey(0);
    212. destroyAllWindows();
    213. }

    补充:
      由于处理空白图像的像素值时没有考虑到最外面的一层,因此会出现有像素保持为初始值0的情况。一种改进的方法是给这些像素赋值为离它最近的像素值,这样的话一定程度上解决了边界全黑的问题。

  • 相关阅读:
    Java中栈
    GoF 23 备忘录模式
    基于MFC和C++的校园导航系统
    运维需要懂产品和运营吗?
    一文速通Nginx网关与gateway网关区分
    Linux命令之chage命令
    6. Design A Web Crawler
    Java继承基础。
    linux进程管理
    武汉新时标文化传媒有限公司抖音小店运营技巧有哪些?
  • 原文地址:https://blog.csdn.net/m0_64007201/article/details/128056274