From 5003f74fba250e224e07501e95922abe603b94e2 Mon Sep 17 00:00:00 2001
From: Tadas Baltrusaitis <tadyla@gmail.com>
Date: Mon, 23 Oct 2017 20:59:54 +0100
Subject: [PATCH] Extra experiments with AUs.

---
 .../helpers/find_Bosphorus.m                  |  4 ++-
 .../run_AU_prediction_BP4D.m                  | 26 +++++++++++--------
 .../run_AU_prediction_Bosphorus.m             |  5 ++++
 3 files changed, 23 insertions(+), 12 deletions(-)

diff --git a/matlab_runners/Action Unit Experiments/helpers/find_Bosphorus.m b/matlab_runners/Action Unit Experiments/helpers/find_Bosphorus.m
index 39c101e..f392bd3 100644
--- a/matlab_runners/Action Unit Experiments/helpers/find_Bosphorus.m	
+++ b/matlab_runners/Action Unit Experiments/helpers/find_Bosphorus.m	
@@ -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
 
diff --git a/matlab_runners/Action Unit Experiments/run_AU_prediction_BP4D.m b/matlab_runners/Action Unit Experiments/run_AU_prediction_BP4D.m
index de807b6..524e2aa 100644
--- a/matlab_runners/Action Unit Experiments/run_AU_prediction_BP4D.m	
+++ b/matlab_runners/Action Unit Experiments/run_AU_prediction_BP4D.m	
@@ -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);
\ No newline at end of file
diff --git a/matlab_runners/Action Unit Experiments/run_AU_prediction_Bosphorus.m b/matlab_runners/Action Unit Experiments/run_AU_prediction_Bosphorus.m
index fff14c5..90e51bc 100644
--- a/matlab_runners/Action Unit Experiments/run_AU_prediction_Bosphorus.m	
+++ b/matlab_runners/Action Unit Experiments/run_AU_prediction_Bosphorus.m	
@@ -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