Welcome 微信登录

首页 / 操作系统 / Linux / 基于HSV色彩空间变换的阴影检测算法——MATLAB实现

图像在不同的色彩空间可以表现出不同的光谱特性,从而适用于不同的应用场合。在 HSV 色彩空间中,阴影在色度和饱和度通道中都与非阴影区域有明显的区别。根据阴影的这种特性,利用连续两次的 HSV 彩色空间变换,可以在图像中有效的区分出阴影与非阴影区域。 一、Otsu阈值算法(Otsu.m)
  1. %利用大津法(Ostu)计算阈值  
  2. %{  
  3. 大津法由大津于1979年提出,对图像Image,记t为前景与背景的分割阈值,前景点数占图像  
  4. 比例为w0,平均灰度为u0;背景点数占图像比例为w1,平均灰度为u1。  
  5. 图像的总平均灰度为:u=w0*u0+w1*u1。从最小灰度值到最大灰度值遍历t,  
  6. 当t使得值g=w0*(u0-u)2+w1*(u1-u)2 最大时t即为分割的最佳阈值。  
  7. 对大津法可作如下理解:该式实际上就是类间方差值,阈值t分割出的前景和背景两部分构成  
  8. 了整幅图像,而前景取值u0,概率为 w0,背景取值u1,概率为w1,总均值为u,根据方差的  
  9. 定义即得该式。因方差是灰度分布均匀性的一种度量,方差值越大,说明构成图像的两部分差  
  10. 别越大, 当部分目标错分为背景或部分背景错分为目标都会导致两部分差别变小,因此使类  
  11. 间方差最大的分割意味着错分概率最小。  
  12. %}  
  13. %Matlab实现大津法计算阀值  
  14. function threshold = Otsu(grayImage)  
  15. %取出最大灰度值,max只能找到每一列的最大值  
  16. maxGrayValue = max(max(grayImage));  
  17. %取出最小灰度值  
  18. minGrayValue = min(min(grayImage));  
  19. %求初始阀值  
  20. t = double(minGrayValue:maxGrayValue);  
  21. %图像大小  
  22. gsize = size(grayImage);  
  23. %图像像素总个数  
  24. muxsize = gsize(1)*gsize(2);  
  25. for k = 1:length(t)  
  26.     %从最小灰度值到最大值的计算方法  
  27.     tk = t(1,k);  
  28.     %定义前景和背景像素数  
  29.     iforeground = 0;  
  30.     ibackground = 0;  
  31.     %定义前景和背景的灰度总和  
  32.     foreground = 0;  
  33.     background = 0;  
  34.     for i = 1:gsize(1)  
  35.         for j = 1:gsize(2)  
  36.             tmp = grayImage(i,j);  
  37.             if(tmp >= tk)  
  38.                 %前景灰度值  
  39.                 iforeground = iforeground + 1;  
  40.                 foreground = foreground + double(tmp);  
  41.             else  
  42.                 %背景灰度值  
  43.                 ibackground = ibackground + 1;  
  44.                 background = background + double(tmp);  
  45.             end  
  46.         end  
  47.     end  
  48.     w0 = iforeground / muxsize;  
  49.     w1 = ibackground / muxsize;  
  50.     u0 = foreground / iforeground;  
  51.     u1 = background / ibackground;  
  52.     t(2,k) = w0*w1*(u0-u1)*(u0-u1);  
  53. end  
  54. %遍历后寻找grayIamge的第二行的最大值  
  55. %第二行方差的最大值,忽略NaN  
  56. omax = max(t(2,:));  
  57. %方差最大值所对应的列号,find只能检查一行中的数,或一维数组  
  58. index = find(t(2,:) >= omax);  
  59. %从第一行取出灰度值作为阀值  
  60. t = t(1,index);  
  61. threshold = t;   
二、阴影检测(ShadowDetection.m)
  1. %清理窗口  
  2. close all  
  3. clear all  
  4. clc  
  5. %读取图像  
  6. img_name = input("请输入图像名字:","s");  
  7. I = imread(img_name);  
  8. J = I;  
  9. %第一次色彩空间变换  
  10. h = rgb2hsv(J);  
  11. %获得图像大小  
  12. [m n] = size(J(:,:,1));  
  13. %将变换后的图像的HSV值规范至[0,255]范围内  
  14. for i = 1:m  
  15.     for j = 1:n  
  16.         for k = 1:3  
  17.             J(i,j,k) = int8(255*h(i,j,k));  
  18.         end  
  19.     end  
  20. end  
  21. %第二次色彩空间变换  
  22. h=rgb2hsv(J);  
  23. %再次将变换后的图像的HSV值规范至[0,255]范围内  
  24. for i = 1:m  
  25.     for j = 1:n  
  26.         for k = 1:3  
  27.             J(i,j,k) = int8(255*h(i,j,k));  
  28.         end  
  29.     end  
  30. end  
  31. %将RGB图像转化为灰度图像  
  32. gray = J(:,:,1);  
  33. %利用Otsu阈值检测法取图像二值化阈值  
  34. t = Otsu(gray);  
  35. %根据阈值将图像二值化  
  36. for i = 1:m  
  37.     for j = 1:n  
  38.         if gray(i,j) < t  
  39.             gray(i,j) = 255;  
  40.         else  
  41.             gray(i,j) = 0;  
  42.         end  
  43.     end  
  44. end  
  45. %利用闭运算去除噪声  
  46. bim=im2bw(gray);  
  47. SE=strel("arbitrary",eye(5));    
  48. BW2=imclose(bim,SE);  
  49. %显示原始图像  
  50. imshow(I);  
  51. title("原始图像");  
  52. %最终结果  
  53. figure,imshow(BW2);  
  54. title("阴影检测结果");