用MATLAB实现对运动物体识别与跟踪

不得不说MATLAB的图像处理函数有点多,但速度有时也是出奇的慢。还是想c的指针,虽然有点危险,但速度那是杠杠的。
第二个MATLAB程序,对运动物体的识别与追踪。
这里我们主要运用帧差法实现运动物体与背景图像的分离,由于视频中的物体较为简单,我们只对两帧图像取帧差(也是为了提高速度)
对于运动物体的提取我们运用了MATLAB里自带的函数bwareaopen
bwareaopen(src,int),src为二值图像,int为设置的联通域的大小,是对帧差法,在转化为二值的图像进行操作,结果是将大小小于设定的int的连通域置为0;
对于第一帧与第二帧图像运动物体的坐标的提取我们用了自带的regionprops函数
regionprops(src,’‘)其中src为传入的二值图像,’‘内的为你所需要的属性
具体属性可以查看MATLAB的help
这里写图片描述
这里我们选用了其中的Centroid属性,返回的时连通域的质心坐标,注返回的第一个值为横坐标,第二个值为纵坐标~
对于运动物体的追踪我们用了质心追踪,
在第一二三两帧的帧间差的运动物体的质心求出来后,将质心做差得到的向量预测下一帧间差运动物体可能到达的位置,接下来在对这个位置进行局部的找质心,再做差如此循环。
追踪大致如图(画的不好):
这里写图片描述
这个相对与全局再次进行bwareaopen,regionprops,速度应该会快一下,而且这是进行局部地搜索所以可以减小背景,或其他噪声的影响。
这里写图片描述
如图为直接进行帧间差分后的转化的二值图像,即使噪声很大用局部追踪也能跟上。
具体代码:

yuandian=zeros(2,2);
i=1293;
filename1 = strcat('I:\2d3d\2\dxshiyan2\C',num2str(i),'.jpg');
src1=imread('C1291.jpg');
src2=imread('C1292.jpg');
%src1=rgb2gray(src1);
%src2=rgb2gray(src2);
zhic1=src2-src1;
zhic1=im2bw(zhic1,0.2);
imshow(zhic1);
zhic2= bwareaopen(zhic1,3000);
imshow(zhic2);
quyu=regionprops(zhic2,'Centroid');
[u,v]=zhixin(zhic2,int16(quyu.Centroid(2)),int16(quyu.Centroid(1)));
yuandian(1,1)=u;
yuandian(1,2)=v;
%去一个模板为400*400src1=src2;
src2=imread('C1293.jpg');
%src2=rgb2gray(src2);
zhic2=src2-src1;
zhic1=im2bw(zhic2,0.2);
zhic2= bwareaopen(zhic1,3000);
imshow(zhic2);
quyu=regionprops(zhic2,'Centroid');
[u,v]=zhixin(zhic2,int16(quyu.Centroid(2)),int16(quyu.Centroid(1)));
yuandian(2,1)=u;
yuandian(2,2)=v;
for i=1294:1386
    filename1 = strcat('I:\2d3d\2\dxshiyan2\C',num2str(i),'.jpg');
    src1=src2;
    src2=imread(filename1);
    src3=src2;
    zhic2=src2-src1;
    zhic2=im2bw(zhic2,0.2); 
    imshow(zhic2);
    houx=2*yuandian(2,2)-yuandian(1,2);
    houy=2*yuandian(2,1)-yuandian(1,1);
    [u,v]=zhixin(zhic2,houy,houx); 
    [state,result]=draw_rect(src3,[v-250,u-250],[500,500],0);
    %imshow(result);
    pause(0.01);
    yuandian(1,1)=yuandian(2,1);
    yuandian(1,2)=yuandian(2,2);
    if(abs(yuandian(1,1)-u<400)&&abs(yuandian(1,2)-v<400))
    yuandian(2,1)=u;
    yuandian(2,2)=v;
    end
end

fanction:

%找质心
function [u,v]=zhixin(erzhi,y,x)
u=0;
v=0;
zuobiaox=0;
zuobiaoy=0;
number=0;
jilux=0;
jiluy=0;
for i=y-200:y+200
    for j=x-200:x+200
        if(j<3839)
        if(erzhi(i,j)==1)
            number=number+1;
            u=(int64(i)+int64(u));
            v=(int64(j)+int64(v));
            zuobiaox=zuobiaox+double(erzhi(i,j));
            zuobiaoy=double(erzhi(i,j))+zuobiaoy;
            jilux=int64(jilux)+int64(j);
            jiluy=int64(jiluy)+int64(i);
        end
        end
    end
end
u=int64(u/zuobiaoy);
%p=jilux/int64(number)-u;
v=int64(v/zuobiaox);
%q=jiluy/int64(number)-v;
%k=0;

draw-rect

function [state,result]=draw_rect(img,startPosition,windowSize,showOrNot)
% 函数调用:[state,result]=draw_rect(img,startPosition,windowSize,showOrNot)
% 函数功能:在图像画个长方形框
% 函数输入:img为原始的大图,可为灰度图,可为彩色图
%          startPosition 框的左上角在大图中的坐标(每行代表x,y坐标),startPosition=[10,30],分别表示x,y为10,30
%          windowSize 框的大小 windowSize=[112,92] 分别表示宽、高
%          showOrNot 是否要显示结果?默认为显示出来?
% 函数输出:state -- 表示程序结果状态?
%          result - 结果图像数据 


if nargin < 4
    showOrNot = 1;
end

rgb = [255 0 0];                                 % 边框颜色
lineSize = 3;                                      % 边框大小,取1,2,3

windowSize(1,1)=windowSize(1,1);
windowSize(1,2) = windowSize(1,2);
if windowSize(1,2) > size(img,1) ||...
        windowSize(1,1) > size(img,2)
    state = -1;                                     % 说明窗口太大,图像太小,
    disp('the window size is larger then image...');
    return;
end

result = img;
if size(img,3) == 3
    for k=1:3
        for i=1:size(startPosition,1) %矩形框的总数
            if(startPosition(i,1)>=0 && startPosition(i,2)>=0)
                result(startPosition(i,2),startPosition(i,1):startPosition(i,1)+windowSize(i,1),k) = rgb(1,k); %画上边框  
                result(startPosition(i,2):startPosition(i,2)+windowSize(i,2),startPosition(i,1)+windowSize(i,1),k) = rgb(1,k);%画右边框
                result(startPosition(i,2)+windowSize(i,2),startPosition(i,1):startPosition(i,1)+windowSize(i,1),k) = rgb(1,k);  %画下边框   
                result(startPosition(i,2):startPosition(i,2)+windowSize(i,2),startPosition(i,1),k) = rgb(1,k);   %画左边框  

                if lineSize == 2 || lineSize == 3
                    result(startPosition(i,2)+1,startPosition(i,1):startPosition(i,1)+windowSize(i,1),k) = rgb(1,k);  
                    result(startPosition(i,2):startPosition(i,2)+windowSize(i,2),startPosition(i,1)+windowSize(i,1)-1,k) = rgb(1,k);
                    result(startPosition(i,2)+windowSize(i,2)-1,startPosition(i,1):startPosition(i,1)+windowSize(i,1),k) = rgb(1,k);
                    result(startPosition(i,2):startPosition(i,2)+windowSize(i,2),startPosition(i,1)-1,k) = rgb(1,k);

                    if lineSize == 3
                        result(startPosition(i,2)-1,startPosition(i,1):startPosition(i,1)+windowSize(i,1),k) = rgb(1,k);   
                        result(startPosition(i,2):startPosition(i,2)+windowSize(i,2),startPosition(i,1)+windowSize(i,1)+1,k) = rgb(1,k);
                        result(startPosition(i,2)+windowSize(i,2)+1,startPosition(i,1):startPosition(i,1)+windowSize(i,1),k) = rgb(1,k);
                        result(startPosition(i,2):startPosition(i,2)+windowSize(i,2),startPosition(i,1)+1,k) = rgb(1,k);
                    end
                end
            end
        end
    end
end

state = 1;

if showOrNot == 1
    figure;
    hold on;
    imshow(result);
end

追踪效果:
这里写图片描述
这里写图片描述
时间:
这里写图片描述

注(画框函数引用网上现有的)
可能是matlab问题,这段视频为4K,无法直接读取,所以只能将其转化为图片逐帧读取。
结果是还是太慢,而且不能预读取,好像用这个无法实时,可能与图片为4k也有点关系有点伤

  • 33
    点赞
  • 251
    收藏
    觉得还不错? 一键收藏
  • 121
    评论
以下是一个简单的MATLAB代码实现,用于运动物体的检测和跟踪: ```matlab %% 读取视频文件 videoReader = VideoReader('input.mp4'); %% 创建视频写入器 videoWriter = VideoWriter('output.avi'); videoWriter.FrameRate = videoReader.FrameRate; open(videoWriter); %% 初始化背景 background = read(videoReader); background_bw = rgb2gray(background); %% 设置参数 threshold = 25; min_object_size = 100; %% 处理每一帧 while hasFrame(videoReader) %% 读取当前帧 frame = readFrame(videoReader); frame_bw = rgb2gray(frame); %% 计算帧差 frame_diff = abs(double(frame_bw) - double(background_bw)); frame_diff = uint8(frame_diff); %% 二值化处理 frame_diff(frame_diff < threshold) = 0; frame_diff(frame_diff >= threshold) = 255; %% 进行形态学操作 se = strel('square', 3); frame_diff = imclose(frame_diff, se); frame_diff = imfill(frame_diff, 'holes'); %% 标识目标位置 [labeled, num_objects] = bwlabel(frame_diff); stats = regionprops(labeled, 'BoundingBox', 'Area'); for i = 1:num_objects if stats(i).Area > min_object_size bbox = stats(i).BoundingBox; rectangle('Position', [bbox(1), bbox(2), bbox(3), bbox(4)], 'EdgeColor', 'r', 'LineWidth', 2); end end %% 导出帧 writeVideo(videoWriter, getframe(gca)); %% 更新背景 alpha = 0.05; background_bw = alpha * double(frame_bw) + (1 - alpha) * double(background_bw); background_bw = uint8(background_bw); end %% 关闭视频写入器 close(videoWriter); ``` 在这个代码中,我们从视频文件中读取每一帧并将其转换为灰度图像。然后我们使用帧差法计算当前帧和背景之间的差异。我们使用阈值来将差异二值化,并使用形态学操作来进一步处理图像。最后,我们使用regionprops函数来标识主要目标的位置,并在视频中导出带有bounding box的帧。我们还使用一个alpha值来更新背景图像。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 121
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值