mirror of
https://gitcode.com/gh_mirrors/ope/OpenFace.git
synced 2025-12-30 13:02:30 +00:00
37 lines
1.6 KiB
Matlab
37 lines
1.6 KiB
Matlab
function [ vid_locs, bboxes, gts_all, invalid_frames ] = CollectTrainData( db_root, bb_root, extra_dir )
|
|
%COLLECTTESTDATA Summary of this function goes here
|
|
% Detailed explanation goes here
|
|
|
|
% all_train = [1,2,3,4,7,9,10,11,13,15,16,17,18,19,20,22,25,27,28,29,31,33,34,35,37,39,41,43,44,46,47,48,49,53,57,59,112,113,115,119,120,123,138,143,144,160,204,205,223,225];
|
|
all_train = [1,2,3,4,7,9,10,11,13,15,16,17,18,19,20,22,25,27,29,31,33,34,35,37,39,41,43,44,46,47,48,49,53,57,59,112,113,115,119,120,123,138,143,144,160,204,205,223,225];
|
|
|
|
vid_locs = cell(numel(all_train),1);
|
|
bboxes = cell(numel(all_train),1);
|
|
gts_all = cell(numel(all_train),1);
|
|
invalid_frames = cell(numel(all_train),1);
|
|
for i=1:numel(all_train)
|
|
id = sprintf('%03d', all_train(i));
|
|
|
|
vid_locs{i} = [db_root '/', id, '/vid.avi'];
|
|
bboxes{i} = dlmread([bb_root, id, '_dets.txt'], ',');
|
|
|
|
%% Grab the ground truth
|
|
fps_all = dir([db_root, '/', id, '/annot/*.pts']);
|
|
gt_landmarks = zeros([68, 2, size(fps_all)]);
|
|
for k = 1:size(fps_all)
|
|
gt_landmarks_frame = dlmread([db_root, '/', id, '/annot/', fps_all(k).name], ' ', 'A4..B71');
|
|
gt_landmarks(:,:,k) = gt_landmarks_frame;
|
|
end
|
|
% Remove unreliable frames
|
|
if(exist([extra_dir, '/', num2str(all_train(i)), '.mat'], 'file'))
|
|
inv_frames = load([extra_dir, '/', num2str(all_train(i)), '.mat']);
|
|
|
|
gt_landmarks(:,:,int32(inv_frames.error)) = [];
|
|
invalid_frames{i} = inv_frames.error;
|
|
end
|
|
gts_all{i} = gt_landmarks;
|
|
end
|
|
|
|
end
|
|
|