sustaining_gazes/matlab_version/bounding_box_mapping/collect_transform_ocv.m

71 lines
1.4 KiB
Matlab

root_test_data = '../../test data/';
[~,dets,gt_labels] = Collect_wild_imgs(root_test_data);
%% Analysing the bounding boxes and errors
detection_root = './openCV_haarcascade_frontalface_alt/';
%%
afw_loc = [detection_root, 'out_haar_afw/'];
afw_dets = dir([afw_loc '*.pts']);
bboxes = [];
bboxes = cat(1, bboxes, zeros(numel(afw_dets), 6));
count = 0;
for i=1:numel(afw_dets)
count = count + 1;
bboxes(count,:) = csvread([afw_loc, afw_dets(i).name]);
end
%%
lfpw_loc = [detection_root, 'out_lfpw_trainset/'];
lfpw_dets = dir([lfpw_loc '*.pts']);
bboxes = cat(1, bboxes, zeros(numel(lfpw_dets), 6));
for i=1:numel(lfpw_dets)
count = count + 1;
bboxes(count,:) = csvread([lfpw_loc, lfpw_dets(i).name]);
end
%%
ibug_loc = [detection_root, 'out_ibug/'];
ibug_dets = dir([ibug_loc '*.pts']);
bboxes = cat(1, bboxes, zeros(numel(ibug_dets), 6));
for i=1:numel(ibug_dets)
count = count + 1;
bboxes(count,:) = csvread([ibug_loc, ibug_dets(i).name]);
end
%%
helen_loc = [detection_root, 'out_helen_trainset/'];
helen_dets = dir([helen_loc '*.pts']);
bboxes = cat(1, bboxes, zeros(numel(helen_dets), 6));
for i=1:numel(helen_dets)
count = count + 1;
% f = fopen([afw_loc, afw_dets(i).name]);
bboxes(count,:) = csvread([helen_loc, helen_dets(i).name]);
% fclose(f);
% bboxes(count,:) = bbox;
end
%%
save('ocv.mat', 'bboxes', 'dets', 'gt_labels');