otb 数据集评估代码注释

来源:CsdnWujinming 发布时间:2019-04-26 15:54:58 阅读量:618

main_running.m

addpath('./util');%添加util路径

addpath(('D:\Document\20190415-0422\vlfeat-0.9.20-bin\vlfeat-0.9.20\toolbox'));%添加vlfeat工具路径

vl_setup%安装vlfeat

addpath(('./rstEval'));%添加rstEval路径

addpath(['./trackers/VIVID_Tracker'])

seqs=configSeqs;%读取测试集序列文件,configSeqs中保存图片位置等信息

trackers=configTrackers;%读取跟踪器信息,包括跟踪器程序入口

shiftTypeSet = {'left','right','up','down','topLeft','topRight','bottomLeft','bottomRight','scale_8','scale_9','scale_11','scale_12'};%变换类型

evalType='OPE'; %'OPE','SRE','TRE'%评估指标

diary(['./tmp/' evalType '.txt']);

numSeq=length(seqs);%视频个数

numTrk=length(trackers);%评估的跟踪器个数 

finalPath = ['./results/results_' evalType '_CVPR13/'];%结果保存目录

if ~exist(finalPath,'dir')

    mkdir(finalPath);

end

%结果目标是否存在

tmpRes_path = ['./tmp/' evalType '/'];

bSaveImage=0;%跟踪的图片是否保存

if ~exist(tmpRes_path,'dir')

    mkdir(tmpRes_path);

end

pathAnno = './anno/';%测试的注释目录

for idxSeq=1:length(seqs)%对于每个视频,调用不同跟踪器

    s = seqs{idxSeq};%保存第i个视频信息

    s.len = s.endFrame - s.startFrame + 1;%视频总帧数

    s.s_frames = cell(s.len,1);%保存视频帧图片地址

    nz = strcat('%0',num2str(s.nz),'d'); %number of zeros in the name of image

    for i=1:s.len%对于视频每一帧

        image_no = s.startFrame + (i-1);%当前帧索引i

        id = sprintf(nz,image_no);

        s.s_frames{i} = strcat(s.path,id,'.',s.ext);%第i帧图片路径

    end

    img = imread(s.s_frames{1});%读取第1帧图片

    [imgH,imgW,ch]=size(img);%图片宽高

    rect_anno = dlmread([pathAnno s.name '.txt']);%当前视频序列的注释文件

    numSeg = 20;

    [subSeqs, subAnno]=splitSeqTRE(s,numSeg,rect_anno);%考虑时间鲁棒性,将图片序列分成20个片段

    switch evalType

        case 'SRE'%评估空间鲁棒性

            subS = subSeqs{1};

            subA = subAnno{1};

            subSeqs=[];

            subAnno=[];

            r=subS.init_rect;

            for i=1:length(shiftTypeSet)

                subSeqs{i} = subS;

                shiftType = shiftTypeSet{i};

                subSeqs{i}.init_rect=shiftInitBB(subS.init_rect,shiftType,imgH,imgW);%将初始框做相应变换

                subSeqs{i}.shiftType = shiftType;%记录变换类型

                subAnno{i} = subA;

            end

        case 'OPE'

            subS = subSeqs{1};

            subSeqs=[];

            subSeqs{1} = subS;

            subA = subAnno{1};

            subAnno=[];

            subAnno{1} = subA;

        otherwise

    end

    for idxTrk=1:numTrk%对所有跟踪器计算结果

        t = trackers{idxTrk};%第t个跟踪器索引

        % validate the results

        if exist([finalPath s.name '_' t.name '.mat'])

            load([finalPath s.name '_' t.name '.mat']);

            bfail=checkResult(results, subAnno);

            if bfail

                disp([s.name ' '  t.name]);

            end

            continue;

        end

        switch t.name

            case {'VTD','VTS'}

                continue;

        end

        results = [];

        for idx=1:length(subSeqs)

            disp([num2str(idxTrk) '_' t.name ', ' num2str(idxSeq) '_' s.name ': ' num2str(idx) '/' num2str(length(subSeqs))])%第idxTrk个跟踪器在第idxSeq个序列上第idx个片段的跟踪

            rp = [tmpRes_path s.name '_' t.name '_' num2str(idx) '/'];%检测结果图片保存路径

            if bSaveImage&~exist(rp,'dir')

                mkdir(rp);

            end

            subS = subSeqs{idx};

            subS.name = [subS.name '_' num2str(idx)];

            funcName = ['res=run_' t.name '(subS, rp, bSaveImage);'];%调用跟踪器命令串

            try

                switch t.name

                    case {'VR','TM','RS','PD','MS'}

                    otherwise

                        cd(['./trackers/' t.name]);

                        addpath(genpath('./'))

                end

                eval(funcName);%解析执行调用跟踪器命令

                switch t.name

                    case {'VR','TM','RS','PD','MS'}

                    otherwise

                        rmpath(genpath('./'))

                        cd('../../');

                end

                if isempty(res)

                    results = [];

                    break;

                end

            catch err

                disp('error');

                rmpath(genpath('./'))

                cd('../../');

                res=[];

                continue;

            end

            res.len = subS.len;

            res.annoBegin = subS.annoBegin;

            res.startFrame = subS.startFrame;

            switch evalType

                case 'SRE'

                    res.shiftType = shiftTypeSet{idx};

            end

            results{idx} = res;

        end

        save([finalPath s.name '_' t.name '.mat'], 'results');%保存结果

    end

end




标签: 数据库
分享:
评论:
你还没有登录,请先