forked from yuxng/MDP_Tracking
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathMOT_display_results.m
58 lines (51 loc) · 1.77 KB
/
MOT_display_results.m
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
% --------------------------------------------------------
% MDP Tracking
% Copyright (c) 2015 CVGL Stanford
% Licensed under The MIT License [see LICENSE for details]
% Written by Yu Xiang
% --------------------------------------------------------
function MOT_display_results
is_save = 1;
opt = globals();
N = numel(opt.mot2d_test_seqs);
seq_set = 'test';
for seq_idx = 1:N
close all;
hf = figure(1);
seq_name = opt.mot2d_test_seqs{seq_idx};
seq_num = opt.mot2d_test_nums(seq_idx);
% build the dres structure for images
filename = sprintf('%s/%s_dres_image.mat', opt.results, seq_name);
if exist(filename, 'file') ~= 0
object = load(filename);
dres_image = object.dres_image;
fprintf('load images from file %s done\n', filename);
else
dres_image = read_dres_image(opt, seq_set, seq_name, seq_num);
fprintf('read images done\n');
save(filename, 'dres_image', '-v7.3');
end
% read tracking results
filename = sprintf('results_MOT/results_MOT_1/%s.txt', seq_name);
dres_track = read_mot2dres(filename);
fprintf('read tracking results from %s\n', filename);
ids = unique(dres_track.id);
cmap = colormap(hsv(numel(ids)));
cmap = cmap(randperm(numel(ids)),:);
if is_save
result_dir = sprintf('results_MOT/results_MOT_1/%s', seq_name);
if exist(result_dir, 'dir') == 0
mkdir(result_dir);
end
end
for fr = 1:seq_num
show_dres(fr, dres_image.I{fr}, '', dres_track, 2, cmap);
if is_save
filename = sprintf('%s/%06d.png', result_dir, fr);
hgexport(hf, filename, hgexport('factorystyle'), 'Format', 'png');
% saveas(hf, filename);
else
pause;
end
end
end