56 lines
No EOL
2.4 KiB
Matlab
56 lines
No EOL
2.4 KiB
Matlab
clear;
|
|
|
|
% fitting parameters more suitable for clnf
|
|
|
|
%%
|
|
% Run the BU test with clnf
|
|
if exist('D:/Datasets/HeadPose', 'file')
|
|
database_root = 'D:/Datasets/HeadPose/';
|
|
elseif(exist([getenv('USERPROFILE') '/Dropbox/AAM/test data/'], 'file'))
|
|
database_root = [getenv('USERPROFILE') '/Dropbox/AAM/test data/'];
|
|
elseif(exist([getenv('USERPROFILE') 'F:/Dropbox/Dropbox/AAM/test data/'], 'file'))
|
|
database_root = 'F:/Dropbox/Dropbox/AAM/test data/';
|
|
elseif(exist('F:/Dropbox/AAM/test data/', 'file'))
|
|
database_root = 'F:/Dropbox/AAM/test data/';
|
|
elseif(exist('/multicomp/datasets/head_pose_dbs', 'file'))
|
|
database_root = '/multicomp/datasets/head_pose_dbs/';
|
|
elseif(exist('/media/tadas/5E08AE0D08ADE3ED/Dropbox/AAM/test data', 'file'))
|
|
database_root = '/media/tadas/5E08AE0D08ADE3ED/Dropbox/AAM/test data';
|
|
end
|
|
|
|
buDir = [database_root, '/bu/uniform-light/'];
|
|
|
|
% The fast and accurate clnf
|
|
%%
|
|
[resFolderBU_OF] = run_bu_experiment(buDir, false, 'model', 'model/main_clnf_general.txt');
|
|
[bu_error_OF, pred_hp_bu, gt_hp_bu, all_errors_bu_OF, rels_bu] = calcBUerror(resFolderBU_OF, buDir);
|
|
|
|
%%
|
|
% Run the Biwi test
|
|
biwi_dir = '/biwi pose/';
|
|
|
|
[res_folder_biwi_OF] = run_biwi_experiment(database_root, biwi_dir, false, 'model', 'model/main_clnf_general.txt');
|
|
% Calculate the resulting errors
|
|
[biwi_error_OF, pred_hp_biwi, gt_hp_biwi, ~, all_errors_biwi_OF, rels_biwi] = calcBiwiError(res_folder_biwi_OF, [database_root biwi_dir]);
|
|
|
|
%% Run the ICT test
|
|
ict_dir = ['/ict/'];
|
|
|
|
% Intensity
|
|
[res_folder_ict_OF] = run_ict_experiment(database_root, ict_dir, false, 'model', 'model/main_clnf_general.txt');
|
|
% Calculate the resulting errors
|
|
[ict_error_OF, pred_hp_ict, gt_hp_ict, ~, all_errors_ict_OF, rel_ict] = calcIctError(res_folder_ict_OF, [database_root ict_dir]);
|
|
|
|
%% Save the results
|
|
filename = 'results/Pose_OF';
|
|
save(filename);
|
|
|
|
% Also save them in a reasonable .txt format for easy comparison
|
|
f = fopen('results/Pose_OF.txt', 'w');
|
|
fprintf(f, 'Dataset and model, pitch, yaw, roll, mean, median\n');
|
|
fprintf(f, 'biwi error: %.3f, %.3f, %.3f, %.3f, %.3f\n', biwi_error_OF, mean(all_errors_biwi_OF(:)), median(all_errors_biwi_OF(:)));
|
|
fprintf(f, 'bu error: %.3f, %.3f, %.3f, %.3f, %.3f\n', bu_error_OF, mean(all_errors_bu_OF(:)), median(all_errors_bu_OF(:)));
|
|
fprintf(f, 'ict error: %.3f, %.3f, %.3f, %.3f, %.3f\n', ict_error_OF, mean(all_errors_ict_OF(:)), median(all_errors_ict_OF(:)));
|
|
|
|
fclose(f);
|
|
clear 'f' |