mirrors
/
person-re-ranking
mirror of https://github.com/zhunzhong07/person-re-ranking
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Update Market_1501_evaluation.m

pull/25/head
zhun zhong 2017-05-05 12:57:07 +08:00 committed by GitHub
parent 7cd1c53c7e
commit 6afa6fa79a
1 changed files with 23 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
evaluation/Market_1501_evaluation.m
View File

@ -14,6 +14,11 @@ addpath(genpath('LOMO_XQDA/'));
run('KISSME/toolbox/init.m');
addpath(genpath('utils/'));
%% re-ranking setting
k1 = 20;
k2 = 6;
lambda = 0.3;
%% network name
netname = 'ResNet_50'; % network: CaffeNet or ResNet_50 googlenet
@ -58,6 +63,16 @@ fprintf(['The IDE (' netname ') + Euclidean performance:\n']);
fprintf(' Rank1, mAP\n');
fprintf('%5.2f%%, %5.2f%%\n\n', CMC_eu(1) * 100, map_eu(1)*100);
%% Euclidean + re-ranking
query_num = size(probFea, 2);
dist_eu_re = re_ranking( [probFea galFea], 1, 1, query_num, k1, k2, lambda);
[CMC_eu_re, map_eu_re, ~, ~] = evaluation(dist_eu_re, label_gallery, label_query, cam_gallery, cam_query);
fprintf(['The IDE (' netname ') + Euclidean + re-ranking performance:\n']);
fprintf(' Rank1, mAP\n');
fprintf('%5.2f%%, %5.2f%%\n\n', CMC_eu_re(1) * 100, map_eu_re(1)*100);
%% train and test XQDA
[train_sample1, train_sample2, label1, label2] = gen_train_sample_xqda(label_train, cam_train, train_feature); % generate pairwise training features for XQDA
[W, M_xqda] = XQDA(train_sample1, train_sample2, label1, label2);% train XQDA
@ -68,3 +83,11 @@ dist_xqda = MahDist(M_xqda, galFea' * W, probFea' * W); % calculate MahDist betw
fprintf(['The IDE (' netname ') + XQDA performance:\n']);
fprintf(' Rank1, mAP\n');
fprintf('%5.2f%%, %5.2f%%\n\n', CMC_xqda(1) * 100, map_xqda(1)*100);
%% XQDA + re-ranking
query_num = size(probFea, 2);
dist_xqda_re = re_ranking( [probFea galFea], M_xqda, W, query_num, k1, k2, lambda);
[CMC_xqda_re, map_xqda_re, ~, ~] = evaluation(dist_xqda_re, label_gallery, label_query, cam_gallery, cam_query);
fprintf(['The IDE (' netname ') + XQDA + re-ranking performance:\n']);
fprintf(' Rank1, mAP\n');
fprintf('%5.2f%%, %5.2f%%\n\n', CMC_xqda_re(1) * 100, map_xqda_re(1)*100);