51 lines
1.5 KiB
Matlab
51 lines
1.5 KiB
Matlab
function [ labels, valid_ids, vid_ids, vid_names ] = extract_SEMAINE_labels( SEMAINE_dir, recs, aus )
|
|
%EXTRACT_SEMAINE_LABELS Summary of this function goes here
|
|
% Detailed explanation goes here
|
|
|
|
% Get the right eaf file
|
|
|
|
aus_SEMAINE = [2 12 17 25 28 45];
|
|
|
|
inds_to_use = [];
|
|
|
|
for i=1:numel(aus)
|
|
|
|
inds_to_use = cat(1, inds_to_use, find(aus_SEMAINE == aus(i)));
|
|
|
|
end
|
|
|
|
labels = cell(numel(recs), 1);
|
|
valid_ids = cell(numel(recs), 1);
|
|
vid_names = cell(numel(recs), 1);
|
|
vid_ids = zeros(numel(recs), 2);
|
|
|
|
for i=1:numel(recs)
|
|
|
|
file = dir([SEMAINE_dir, '/', recs{i}, '/*.eaf']);
|
|
|
|
vid_ids(i,:) = dlmread([SEMAINE_dir, '/', recs{i}, '.txt'], ' ');
|
|
|
|
vid_names_c = dir([SEMAINE_dir, '/', recs{i}, '/*.avi']);
|
|
[~, vid_names{i},~] = fileparts(vid_names_c.name);
|
|
|
|
xml_file = [SEMAINE_dir, recs{i}, '\' file.name];
|
|
[root_xml, name_xml, ~] = fileparts(xml_file);
|
|
|
|
activations = ParseSEMAINEAnnotations([SEMAINE_dir, recs{i}, '/' file.name]);
|
|
|
|
if(size(activations,1) < vid_ids(i,2))
|
|
vid_ids(i,2) = size(activations,1);
|
|
if(vid_ids(i,2) > 2999)
|
|
vid_ids(i,1) = vid_ids(i,2) - 2999;
|
|
end
|
|
end
|
|
|
|
labels{i} = activations(vid_ids(i,1)+1:vid_ids(i,2), 1 + inds_to_use);
|
|
|
|
% all indices in SEMAINE are valid
|
|
valid_ids{i} = ones(size(labels{i},1),1);
|
|
|
|
end
|
|
|
|
end
|
|
|