用Matlab实现路口车量的计数

2019-04-14 22:00发布

最近有一个作业,用matlab写一段代码来检测路口的车流量(虽然我一直喜欢用C胜过用matlab,原因你懂的,但是这是作业没办法),简单实现了下。 这里写图片描述
这是视频中的一帧图片,可以看到主要分为4个车道(应该有5个)由于视频是直行时拍的所以不考虑最右边的转弯车道,也不考虑自行车等非机动车。
首先这边使用背景差分法,由于视频的时间比较短,笔者只选用了没有车的一张图片作为背景,如果要进行长时间的计算,应该进行背景的更新,效果应该会更好一些。背景更新,可以用均值滤波法,W4法或者一些背景统计的算法。
将背景转化为灰度,;
在选取另一张图片,转化为灰度图,与原来背景灰度图进行相减,可以得到一张背景差分图片,再将背景差分图转化为二值图像,在转化为二值图像时用合适的阈值将车道上的噪声消除(有车时几乎没有噪声,无车时噪声出奇大,这可能与摄像头有关),这里转换出来的车可能被分割为大大小小多块,笔者用的时膨胀和腐蚀,将小的腐蚀掉或用膨胀将大块和小块相连接。 对整个机动车道按车道分成四部分: 这里写图片描述 可以用matlab中figure上的数据游标大致的测量出车道(图中的黑线)在图片中坐标位置,大致计算出黑线的函数表达式,方便对车道的切割。 如图: 这里写图片描述
这是笔者对第一车道的分割结果。
主要代码要根据你的黑线函数式: for u=1:a(1,1) for v=1:a(1,2) if(v>(u-287)/(-1.6)&&v<((u-843)/(-3))) dst1(u,v,1)=erzhi(u,v,1); elseif(v>((u-843)/(-3))&&v<(u+4551)/(13)) dst2(u,v,1)=erzhi(u,v,1); elseif(v<((u+1327)/(3.2))&&v>(u+4551)/(13)) dst3(u,v,1)=erzhi(u,v,1); elseif(v>((u+1327)/(3.2))&&v<(u+1006)/(2)) dst4(u,v,1)=erzhi(u,v,1); end end 这是笔者的分割,主要是将相应坐标像素复制到了四张新的图片中(额~,现在只是在分析,这种复制的方法只适合在分析时使用,实际复制出奇的慢) 对每一个车道进行计数。 对车的计数方法有很多种,可以选择线计数,可以选择块计数,点计数等等,这里我使用块计数,即选取某一块区域当区域中白点(二值化后车的部分),到达一定数目时表明这是有车。由于车速比较慢,不可能在一帧的时间间隔内完全驶过这个区域,所以我将区域由黑变白的时刻当成一辆车,即设置一个标志,记录这时是否有车,与记录上一帧是否有车的标志相减,如果为1,则该车道的车数加一。 代码: for u=261:281 for v=50:70 if(erzhi(u,v)==1) mianji=mianji+1; end end end if(mianji>300) baocun(1,2)=1; else baocun(1,2)=0; end if(baocun(1,2)-baocun(1,1)==1) ceshu(1)=1; else ceshu(1)=0; end jishu(1)=jishu(1)+ceshu(1); baocun(1,1)=baocun(1,2); mianji=0; 如代码中的261到281,50到70是我选择计算的区域;
关于区域的选择,我认为也是有一定的规律的:
1.首先靠内车的几个车道由于中间没有白线,进入入口后,车辆行驶可能会有一定的变道,或者偏离原来的方相,选取地区应尽量靠近入口出,即图片的上半部分,同时应进来靠近车道的中心位置。
2.对于外车道因为有右转弯车道,车辆偏离方向的影响较小。
3.从图片中看摄像头摆放的位置在靠近内车道,外车道的车的光线反射对摄像头的影响大,还有挡风玻璃在二值时会把车切成两半,这时可能要一点膨胀,或连通域。
分析时的总代码: src1=imread('jiebai1.jpg'); src1=rgb2gray(src1); a=size(src1); erzhi=zeros(a(1,1),a(1,2),1); jishu=zeros(4,1); baocun=zeros(4,2); biaoji=zeros(2,1); ceshu=zeros(4,1); dst1=zeros(a(1,1),a(1,2),1); dst2=zeros(a(1,1),a(1,2),1); dst3=zeros(a(1,1),a(1,2),1); dst4=zeros(a(1,1),a(1,2),1); i=1; for i=1:2:216 filename='jiebai'; src2=imread([filename,num2str(i),'.jpg']); src2=rgb2gray(src2); dst_2=src2-src1; dstgray=dst_2; erzhi=im2bw(dstgray,0.25); for u=1:a(1,1) for v=1:a(1,2) if(v>(u-287)/(-1.6)&&v<((u-843)/(-3))) dst1(u,v,1)=erzhi(u,v,1); elseif(v>((u-843)/(-3))&&v<(u+4551)/(13)) dst2(u,v,1)=erzhi(u,v,1); elseif(v<((u+1327)/(3.2))&&v>(u+4551)/(13)) dst3(u,v,1)=erzhi(u,v,1); elseif(v>((u+1327)/(3.2))&&v<(u+1006)/(2)) dst4(u,v,1)=erzhi(u,v,1); end end end %-------------------------------对车道1 baocun(1,2)=chedao1(dst1); if(baocun(1,2)-baocun(1,1)==1) ceshu(1)=1; else ceshu(1)=0; end jishu(1)=jishu(1)+ceshu(1); baocun(1,1)=baocun(1,2); %-------------------------------对车道2 baocun(2,2)=chedao2(dst2); if(baocun(2,2)-baocun(2,1)==1) ceshu(2)=1; else ceshu(2)=0; end jishu(2)=jishu(2)+ceshu(2); baocun(2,1)=baocun(2,2); %-----------------------------对车道3 baocun(3,2)=chedao3(dst3); if(baocun(3,2)-baocun(3,1)==1) ceshu(3)=1; else ceshu(3)=0; end jishu(3)=jishu(3)+ceshu(3); baocun(3,1)=baocun(3,2); %------------------------------对车道4 baocun(4,2)=chedao4(dst4); if(baocun(4,2)-baocun(4,1)==1) ceshu(4)=1; else ceshu(4)=0; end jishu(4)=jishu(4)+ceshu(4); baocun(4,1)=baocun(4,2); pause(0.02); end zhongshu=jishu(1)+jishu(2)+jishu(3)+jishu(4); zhongshu 其中一个子函数: function biaoji=chedao1(erzhi) baocun=0; pengzhuang=[0,1,0;1,1,1;0,1,0]; erzhi=imdilate(erzhi,pengzhuang); erzhi=imdilate(erzhi,pengzhuang); erzhi=imdilate(erzhi,pengzhuang); erzhi=imdilate(erzhi,pengzhuang); erzhi=imdilate(erzhi,pengzhuang); erzhi=imdilate(erzhi,pengzhuang); erzhi=imdilate(erzhi,pengzhuang); erzhi=imdilate(erzhi,pengzhuang); erzhi=imdilate(erzhi,pengzhuang); erzhi=imdilate(erzhi,pengzhuang); erzhi=imdilate(erzhi,pengzhuang); erzhi=imdilate(erzhi,pengzhuang); erzhi=imdilate(erzhi,pengzhuang); erzhi=imdilate(erzhi,pengzhuang); erzhi=imdilate(erzhi,pengzhuang); erzhi=imdilate(erzhi,pengzhuang); imshow(erzhi); for u=261:281 for v=50:70 if(erzhi(u,v)==1) baocun=baocun+1; end end end if(baocun>300) biaoji=1; else biaoji=0; end % if(biaoji(2)-biaoji(1)==1) % jishu=1; % else % jishu=0; % end % biaoji(1)=biaoji(2); %baocun(2)=0; %imshow(erzhi); end 这只是分析,面临的问题时时间太慢,无法进行实时记数。
解决方法时对图片不进行复制为4张图片(复制大概花费了进3/4的时间),直接对视频进行读帧,进行实时处理:
大致代码: videofilename='C:UsersAdministratorDesktopmatlab6jiebai.avi'; video=VideoReader(videofilename); frame_number=floor(video.Duration*video.FrameRate); src1=readFrame(video); src1=rgb2gray(src1); a=size(src1); erzhi=zeros(a(1,1),a(1,2),1); jishu=zeros(4,1); baocun=zeros(4,2); biaoji=zeros(4,2); ceshu=zeros(4,1); baise=zeros(4,1); i=1; pengzhuang=[1,1,1;1,1,1;1,1,1]; for b=2:frame_number ...... src2=[]; end zhongshu=jishu(1)+jishu(2)+jishu(3)+jishu(4); zhongshu matlab用了没想象中的差,还是能勉强进行实时计数,如果想减少时间,可以进行跳帧处理。
对车流量的其他操作可以看接下来的两篇博客