Matlab Code for Visualize the Tracking Results of OTB100 dataset

Matlab Code for Visualize the Tracking Results of OTB100 dataset

2018-11-12 17:06:21

 %把所有tracker的结果画在一张图上,结果保存在当前目录下的trackingResultsDisplay下
 clc; close all; clear all;
 dataPath = 'C:\Users\WANG XIAO\Desktop\Tracking_evaluation\OTB100_benchmark\Benchmark\';
 trackerResultsPath='C:\Users\WANG XIAO\Desktop\Tracking_evaluation\PlotErr_OTBdataset\BBresults\';
 sequencePath=dataPath;
 saveBasePath='C:\Users\WANG XIAO\Desktop\demo_tracking_results\';
 sequences=dir(dataPath);
 sequences=sequences(3:end);
 sequences={sequences.name};

 tracker={   'CSRDCF',   'SRDCF',   'ECO',   'CCOT',   'pyMDNetBaseline' };

 % edgeColor={'r',   'g',    'b', 'y',   'k',   'm',     'c',      'g',    'b',   'y',   'k',      'm'};
 edgeColor(:,:,1)=[1,0,0];           edgeColor(:,:,2)=[0,0,1];       edgeColor(:,:,3)=[0,1,0];
 edgeColor(:,:,4)=[0,1,1];           edgeColor(:,:,5)=[1,0,1];       edgeColor(:,:,6)=[0,0,0];
 edgeColor(:,:,7)=[1,1,1];           

 edgeColor(:,:,8)=[0,1,0];       edgeColor(:,:,9)=[0,1,1];
 edgeColor(:,:,10)=[1,0,1];         edgeColor(:,:,11)=[0,0,0];    edgeColor(:,:,12)=[1,0.5,0];
 edgeColor(:,:,13)=[0.5,0.5,0];  edgeColor(:,:,14)=[0,0,1];    edgeColor(:,:,15)=[0,1,0];
 lineStyle={'-','-','-','-','-','-','-',};

 %% ####################################################
 seqIndexList = {51, 64, 72};   %% set the video index you want to shown. 

 for seqIndex=1:length(seqIndexList)
     trackerResult=[];
     sequence=sequences{seqIndexList{seqIndex}};

     if(isdir(saveBasePath)==0),
         mkdir(saveBasePath);
     end

     savingPath=[saveBasePath sequence '/'];
     if(isdir(savingPath)==0),
         mkdir(savingPath);
         mkdir([savingPath 'v/']);
         mkdir([savingPath 'i/']);

     end
     savingPath; 

     for trackerIndex=1:length(tracker),
         try
             trackerResult(:,:,trackerIndex)=dlmread([trackerResultsPath  tracker{trackerIndex} '_' sequence '.txt']);
         catch
             trackerResult(:,:,trackerIndex)=dlmread([trackerResultsPath  sequence '_' tracker{trackerIndex} '.txt']);
         end

     end

     frames_v=dir([sequencePath sequence '/img/*.jpg']);
     frames_i=dir([sequencePath sequence '/img/*.jpg']);
     if(isempty(frames_v)==1),
         frames_v=dir([sequencePath sequence '/img/*.jpg']);
     end

     if(isempty(frames_i)==1),
         frames_i=dir([sequencePath sequence '/img/*.jpg']);
     end

     frames_v={frames_v.name};
     frames_i={frames_i.name};

     bb=[   trackerResult(:,1,:),  trackerResult(:,2,:) , trackerResult(:,3,:), trackerResult(:,4,:)   ]   ;

     % for the visible images
     for frameIndex=1:length(frames_v),
         im=imread([sequencePath sequence '/img/' frames_v{frameIndex}]);
         imshow(uint8(im));
         for trackerIndex=1:length(tracker),
             if ~isempty(strfind(tracker{trackerIndex},'_i'))==1,                  continue;
             end
             tracker{trackerIndex}

             disp(['==>> frameIndex: ', num2str(frameIndex), '     ==>> trackerIndex: ', num2str(trackerIndex)]);
             disp(['==>> bb: ', num2str(size(bb))]);
             bbtemp=bb(frameIndex,:,trackerIndex);

             if bbtemp(3)<=0,
                 bbtemp(3)=1;
                 bb(frameIndex,:,trackerIndex)=bbtemp;
             end
             if bbtemp(4)<=0,
                 bbtemp(4)=1;
                 bb(frameIndex,:,trackerIndex)=bbtemp;
             end

             if(bb(frameIndex,1,trackerIndex)>0||bb(frameIndex,2,trackerIndex)>0||...
                     bb(frameIndex,3,trackerIndex)>0||bb(frameIndex,4,trackerIndex)>0),
                 rectangle('Position',bb(frameIndex,:,trackerIndex),'LineWidth',2,'EdgeColor',edgeColor(:,:,trackerIndex),'LineStyle',lineStyle{trackerIndex});
             end
         end

         hold on;
         text(10, 30, strcat('#',num2str(frameIndex)), 'Color','y', 'FontWeight','bold', 'FontSize',30);
         set(gca,'position',[0 0 1 1]);
         pause(0.00001);
         hold off;
         imgName=sprintf('%04d.jpg',frameIndex);
         saveas(gca,[savingPath 'v/'  imgName]);
         t1=imread([savingPath 'v/'  imgName]);
         t1=imresize(t1,[272 640]);
 %         imwrite(t1, [savingPath 'v/'  imgName]);

     end

 end