基于matlab行驶车辆检测计数:读入待处理视频,用于色彩空间转换、检测出包含汽车运动的图像块、用高斯混合模型检测背景、对检测出的运动车辆进行框画、标注检测到车辆的个数用于显示结果。
1 案例背景
2 理论基础
2.1 光流法检测运动原理
2.2 光流法的主要方法
clear;close all; SE = strel('rectangle',[10 10]); %注意:结构元素必须具有适当的大小 BW1=imread('test5.jpg'); subplot(221);imshow(BW1);title('原图'); %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%添加 GRAY=rgb2gray(BW1); %灰度图像1 h=fspecial('average',9); GRAY=uint8(round(filter2(h,GRAY))); %进行均值滤波 subplot(222);imshow(GRAY);title('均值滤波'); GRAY = imcrop(GRAY,[10 20 542 355]); %裁剪 GRAY=double(GRAY); %转换为double类型 [Gx,Gy]=gradient(GRAY); % 计算梯度 G=sqrt(Gx.*Gx+Gy.*Gy); % 水平垂直差分 J=GRAY; K=find(G<2); %指定灰度级 J(K)=0; subplot(223);imshow(J);title('锐化'); %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% BW2=imdilate(J,SE);%膨胀 BW3 = imfill(BW2,'holes'); %填充 BW4=imerode(BW3,SE);%腐蚀 I=im2bw(BW4); subplot(224);imshow(I);title('目标块'); %====形态特征值计算===% A=bwarea(I); % 计算目标的面积 I1=edge(I,'canny',0.7); [L, num]=bwlabel(I1,8);%存放提取的目标矩阵和数目 STATS = regionprops(L,'BoundingBox'); % 存放经过筛选以后得到的所有矩形块 n = 1; wh=zeros(1,num); for i=1:num box = STATS(i).BoundingBox; %图像的长宽 x = box(1); %矩形坐标x y = box(2); %矩形坐标y w = box(3); %矩形宽度w h = box(4); %矩形高度h wh(1,i)=w*h; ux = ceil(x); %取整 uy = ceil(y); if ux > 1 ux = ux - 1; end if uy > 1 end %result(n,:) = [ux uy w h]; R=zeros(h,w); G=zeros(h,w); B=zeros(h,w); for j=1:w for ii=1:h R(ii,j)=BW1(ii+uy-1,j+ux-1,1); B(ii,j)=BW1(ii+uy-1,j+ux-1,3); end end % r(1,n)=mean(mean(R)) % g(1,n)=mean(mean(G)) % b(1,n)=mean(mean(B)) % n = n+1; if(r(1,i)>130&&g(1,i)<130&&b(1,i)<130) disp('检测出车辆是红色') elseif(r(1,i)<130&&g(1,i)>130&&b(1,i)<130) disp('检测出车辆是绿色') elseif(r(1,i)<130&&g(1,i)<130&&b(1,i)>130) disp('检测出车辆是蓝色') elseif(r(1,i)<80&&g(1,i)<80&&b(1,i)<80) disp('检测出车辆是黑色') elseif(r(1,i)>170&&g(1,i)>170&&b(1,i)>170) disp('检测出车辆是白色') else disp('检测出车辆是其他色') end end % 创建系统对象,用于显示结果 sz = get(0,'ScreenSize'); pos = [20 sz(4)-300 200 200]; hVideoOrig = vision.VideoPlayer('Name', 'Original', 'Position', pos); pos(1) = pos(1)+220; %在右侧建立下一个视窗 hVideoFg = vision.VideoPlayer('Name', 'Foreground', 'Position', pos); pos(1) = pos(1)+220; hVideoRes = vision.VideoPlayer('Name', 'Results', 'Position', pos); line_row = 23; % 定义感兴趣区域(ROI) % 以下的程序段为对输入的视频图像进行处理 while ~isDone(hvfr) image = step(hvfr); % 读入视频的每一帧 y = step(hcsc, image); % 将彩色图像转换成灰度图像 % 采用自动白平衡算法去除灰度突变 y = y-mean(y(:)); fg_image = step(hfdet, y); % 检验背景 % 采用数学形态学,在前景图像中检测变化的联通图像块区域的面积 [area, bbox] = step(hblob, fg_image); image_out = image; end
完整代码添加QQ1575304183