Проблемы с инициализацией алгоритма CamShift

Question

Я использую Matlab для реализации трекера, используя кулачковый сдвиг. Мне нужно реализовать это самостоятельно, поэтому я не использую встроенный метод.

Я позволил пользователю выбрать объект для отслеживания, и после первой итерации конвергенция заставляет его начать отслеживать другой объект. картинки говорят громче слов так:

Я выбираю свое лицо в первом кадре:

Он приближается к моему плечу и начинает отслеживать его:

Теперь отслеживание в целом нормально во всем видео, поэтому я думаю, что проблема находится на этапе инициализации.

SomeCode: получить позицию от пользователя:

function [pos] = getObjectPosition(F)
f = figure;
imshow(F);
pos = getPosition(imrect(gca));
close(f);
end

Основной цикл:

firstFrame = readFrame(input);
pos = getObjectPosition(firstFrame);

while hasFrame(input)
% init next frame
nextFrame = readFrame(input);
[currFrameHSV, ~, ~] = rgb2hsv(nextFrame);
frameHue = double(currFrameHSV(:,:,1));

% init while loop vars 
prevX = -1;
prevY = -1;
i = 0;
while (i < numOfIterations) && ~(prevX == pos(1) && prevY == pos(2))
    % updating the curr iteration number
    i = i + 1;
    % getting search window location and dimensions
    envRect = round(getEnvRect(pos,pos(4)/yRatio,pos(3)/xRatio,height,width));

    [X,Y] = meshgrid(envRect(1):envRect(1) + envRect(3),...
                     envRect(2):envRect(2) + envRect(4));

    % getting the sub image
    I = imcrop(frameHue,envRect);

    M00 = sum(sum(I));
    M10 = sum(sum(X.*I));
    M01 = sum(sum(Y.*I));

    % getting center mass location
    Xc = round(M10/M00);
    Yc = round(M01/M00);

    % updating object position
    prevX = pos(1);
    prevY = pos(2);
    pos(1) = floor(Xc - pos(3)/2);
    pos(2) = floor(Yc - pos(4)/2);

end

% adding the object's bounding box to the frame to write
% show the image
end

метод getEnv:

function [ envRect ] = getEnvRect( feature,hShift,wShift,rows,cols )
hEnvSize = feature(4) + 2*hShift; 
wEnvSize = feature(3) + 2*wShift;
xPatch  = feature(1);
xEnv    = max(1, xPatch - wShift);
yPatch  = feature(2);
yEnv    = max(1, yPatch - hShift);
width   = min(cols, xEnv + wEnvSize) - xEnv;
height  = min(rows, yEnv + hEnvSize) - yEnv;

envRect = [xEnv, yEnv, width, height];
end