Extra experiments with AUs.
This commit is contained in:
parent
c9061824c5
commit
5003f74fba
3 changed files with 23 additions and 12 deletions
|
@ -1,6 +1,8 @@
|
|||
if(exist('D:/Datasets/Bosphorus/', 'file'))
|
||||
Bosphorus_dir = 'D:\Datasets\Bosphorus/';
|
||||
else
|
||||
elseif(exist('E:/Datasets/Bosphorus/', 'file'))
|
||||
Bosphorus_dir = 'E:\Datasets\Bosphorus/';
|
||||
else
|
||||
fprintf('Bosphorus dataset location not found (or not defined)\n');
|
||||
end
|
||||
|
||||
|
|
|
@ -22,32 +22,32 @@ new_bp4d_dirs = {};
|
|||
% This might take some time
|
||||
for i = 1:numel(bp4d_dirs)
|
||||
dirs = dir([bp4d_loc, '/', bp4d_dirs{i}, '/T*']);
|
||||
|
||||
|
||||
tmp_dir = [bp4d_loc, '/../tmp/', bp4d_dirs{i}, '/'];
|
||||
new_bp4d_dirs = cat(1, new_bp4d_dirs, tmp_dir);
|
||||
|
||||
|
||||
if(~exist(tmp_dir, 'file'))
|
||||
mkdir(tmp_dir);
|
||||
|
||||
|
||||
% Move all images and resize them
|
||||
for d=1:numel(dirs)
|
||||
|
||||
|
||||
in_files = dir([bp4d_loc, '/', bp4d_dirs{i}, '/', dirs(d).name, '/*.jpg']);
|
||||
|
||||
|
||||
for img_ind=1:numel(in_files)
|
||||
|
||||
|
||||
img_file = [bp4d_loc, '/', bp4d_dirs{i}, '/', dirs(d).name, '/', in_files(img_ind).name];
|
||||
img = imread(img_file);
|
||||
img = imresize(img, 0.5);
|
||||
img_out = [tmp_dir, dirs(d).name, '_', in_files(img_ind).name];
|
||||
imwrite(img, img_out);
|
||||
|
||||
|
||||
end
|
||||
|
||||
|
||||
end
|
||||
|
||||
|
||||
end
|
||||
|
||||
|
||||
end
|
||||
%%
|
||||
|
||||
|
@ -125,6 +125,8 @@ end
|
|||
|
||||
%%
|
||||
f = fopen('results/BP4D_valid_res_class.txt', 'w');
|
||||
f1s_class = zeros(1, numel(aus_BP4D));
|
||||
|
||||
for au = 1:numel(aus_BP4D)
|
||||
|
||||
if(inds_au_class(au) ~= 0)
|
||||
|
@ -137,7 +139,7 @@ for au = 1:numel(aus_BP4D)
|
|||
recall = tp./(tp+fn);
|
||||
|
||||
f1 = 2 * precision .* recall ./ (precision + recall);
|
||||
|
||||
f1s_class(au) = f1;
|
||||
fprintf(f, 'AU%d class, Precision - %.3f, Recall - %.3f, F1 - %.3f\n', aus_BP4D(au), precision, recall, f1);
|
||||
end
|
||||
|
||||
|
@ -195,8 +197,10 @@ end
|
|||
|
||||
%%
|
||||
f = fopen('results/BP4D_valid_res_int.txt', 'w');
|
||||
ints_cccs = zeros(1, numel(aus_BP4D));
|
||||
for au = 1:numel(aus_BP4D)
|
||||
[ accuracies, F1s, corrs, ccc, rms, classes ] = evaluate_au_prediction_results( preds_all_int(valid_ids, inds_au_int(au)), labels_gt(valid_ids,au));
|
||||
ints_cccs(au) = ccc;
|
||||
fprintf(f, 'AU%d results - rms %.3f, corr %.3f, ccc - %.3f\n', aus_BP4D(au), rms, corrs, ccc);
|
||||
end
|
||||
fclose(f);
|
|
@ -100,6 +100,7 @@ end
|
|||
f = fopen('results/Bosphorus_res_class.txt', 'w');
|
||||
labels_gt_bin = labels_gt;
|
||||
labels_gt_bin(labels_gt_bin > 1) = 1;
|
||||
f1s_class = zeros(1, numel(aus_Bosph));
|
||||
for au = 1:numel(aus_Bosph)
|
||||
|
||||
tp = sum(labels_gt_bin(:,au) == 1 & labels_pred(:, au) == 1);
|
||||
|
@ -111,6 +112,7 @@ for au = 1:numel(aus_Bosph)
|
|||
recall = tp./(tp+fn);
|
||||
|
||||
f1 = 2 * precision .* recall ./ (precision + recall);
|
||||
f1s_class(au) = f1;
|
||||
|
||||
fprintf(f, 'AU%d class, Precision - %.3f, Recall - %.3f, F1 - %.3f\n', aus_Bosph(au), precision, recall, f1);
|
||||
|
||||
|
@ -180,10 +182,13 @@ end
|
|||
|
||||
%%
|
||||
f = fopen('results/Bosphorus_res_int.txt', 'w');
|
||||
cccs_reg = zeros(1, numel(aus_Bosph));
|
||||
for au = 1:numel(aus_Bosph)
|
||||
|
||||
[ ~, ~, corrs, ccc, rms, ~ ] = evaluate_regression_results( labels_pred(:, au), labels_gt(:, au));
|
||||
|
||||
cccs_reg(au) = ccc;
|
||||
|
||||
fprintf(f, 'AU%d intensity, Corr - %.3f, RMS - %.3f, CCC - %.3f\n', aus_Bosph(au), corrs, rms, ccc);
|
||||
|
||||
end
|
||||
|
|
Loading…
Reference in a new issue