meanshift.m

来自「利用meanshift在ＭＡＴＬＡＢ中实现目标检测和跟踪」· M 代码 · 共 51 行

function [ rowcenter colcenter M00 ] = meanshift(I, rmin, rmax, cmin,...
    cmax, probmap)
%inputs
%   rmin, rmax, cmin, cmax are the coordiantes of the window
%   I is the image
%outputs
%   colcenter rowcenter are the new center coordinates
%   Moo is the zeroth mean

% **********************************************************************
% initialize
% **********************************************************************

M00 = 0; %zeroth mean
M10 = 0; %first moment for x
M01 = 0; %first moment for y
histdim = (0:1:255); % dimensions of histogram... 0 to 255, increment by 1
[rows cols] = size(I);
cols = cols/3; % **********************8

% **********************************************************************
% Main code
% **********************************************************************

% determine zeroth moment
for c = cmin:cmax
    for r = rmin:rmax
        M00 = M00 + probmap(r, c);
    end
end

% determine first moment for x(col) and y(row)
for c = cmin:cmax
    for r = rmin:rmax
        M10 = M10 + c*probmap(r,c);
        M01 = M01 + r*probmap(r,c);
    end
end

% determine new centroid
% x is cols

    colcenter = M10/M00;

% y is rows

    rowcenter = M01/M00;