mosse代码阅读

Fashioneight

已于 2022-09-20 22:20:27 修改

阅读量601

点赞数 1

分类专栏：相关滤波论文文章标签：计算机视觉 matlab 图像处理

于 2022-09-20 22:17:20 首次发布

本文链接：https://blog.csdn.net/Dwyanesyl/article/details/126962806

版权

相关滤波论文专栏收录该内容

2 篇文章 2 订阅

订阅专栏

matlab版代码下载地址：amoudgl/mosse-tracker: Implementation of MOSSE tracker in MATLAB: Visual Object Tracking using Adaptive Correlation Filters (CVPR 2010) (github.com)

对应论文看，其实代码很简单。

主要代码：

mosse.m:

% get images from source directory
datadir = '../data/';
dataset = 'toy1';
path = [datadir dataset];
img_path = [path '/img/'];
D = dir([img_path, '*.jpg']);
seq_len = length(D(not([D.isdir])));
if exist([img_path num2str(1, '%04i.jpg')], 'file'),
    img_files = num2str((1:seq_len)', [img_path '%04i.jpg']);
else
    error('No image files found in the directory.');
end

% select target from first frame
im = imread(img_files(1,:));
f = figure('Name', 'Select object to track'); imshow(im);
rect = getrect;
close(f); clear f;
center = [rect(2)+rect(4)/2 rect(1)+rect(3)/2];              %中心点位置

% plot gaussian
sigma = 100;
gsize = size(im);
[R,C] = ndgrid(1:gsize(1), 1:gsize(2));                             
g = gaussC(R,C, sigma, center);                              %计算选框与周围目标之间临近距离
g = mat2gray(g);                                             %把g变成0-1范围之内的数

% randomly warp original image to create training set
if (size(im,3) == 3) 
    img = rgb2gray(im);                                      %把RGB图像变为灰度图像
end
img = imcrop(img, rect);
g = imcrop(g, rect);
G = fft2(g);
height = size(g,1);
width = size(g,2);
fi = preprocess(imresize(img, [height width]));              %对边界预处理
Ai = (G.*conj(fft2(fi)));
Bi = (fft2(fi).*conj(fft2(fi)));
N = 128;                                                            
for i = 1:N                    %这个N是what？？？？扭曲初始帧获得128个第一帧图像？ 论文里是8个
    fi = preprocess(rand_warp(img));
    Ai = Ai + (G.*conj(fft2(fi)));
    Bi = Bi + (fft2(fi).*conj(fft2(fi)));
end

% MOSSE online training regimen
eta = 0.125;
fig = figure('Name', 'MOSSE');
mkdir(['results_' dataset]);
for i = 1:size(img_files, 1)
    img = imread(img_files(i,:));
    im = img;
    if (size(img,3) == 3)
        img = rgb2gray(img);          %如果输入的是RGB图像，把他转化为灰度图像 hw&3 -》 hw*1
    end
    if (i == 1)
        Ai = eta.*Ai;
        Bi = eta.*Bi;
    else
        Hi = Ai./Bi;
        fi = imcrop(img, rect); 
        fi = preprocess(imresize(fi, [height width]));
        gi = uint8(255*mat2gray(ifft2(Hi.*fft2(fi))));
        maxval = max(gi(:));
        [P, Q] = find(gi == maxval);
        dx = mean(P)-height/2;
        dy = mean(Q)-width/2;
        
        rect = [rect(1)+dy rect(2)+dx width height];
        fi = imcrop(img, rect); 
        fi = preprocess(imresize(fi, [height width]));
        Ai = eta.*(G.*conj(fft2(fi))) + (1-eta).*Ai;
        Bi = eta.*(fft2(fi).*conj(fft2(fi))) + (1-eta).*Bi;
    end
    
    % visualization
    text_str = ['Frame: ' num2str(i)];
    box_color = 'green';
    position=[1 1];
    result = insertText(im, position,text_str,'FontSize',15,'BoxColor',...          
                     box_color,'BoxOpacity',0.4,'TextColor','white');
    result = insertShape(result, 'Rectangle', rect, 'LineWidth', 3);            
                                                                    %加文字和跟踪目标的边框
    imwrite(result, ['results_' dataset num2str(i, '/%04i.jpg')]);
    imshow(result);
end