mirror of
https://gitcode.com/gh_mirrors/ope/OpenFace.git
synced 2026-01-08 17:30:18 +00:00
98 lines
2.6 KiB
Matlab
98 lines
2.6 KiB
Matlab
function Script_CECLM_cross_data()
|
|
|
|
addpath(genpath('../'));
|
|
|
|
[images, detections, labels] = Collect_menpo_imgs('G:\Datasets\menpo/');
|
|
|
|
%% loading the CE-CLM model and parameters
|
|
[patches, pdm, clmParams, early_term_params] = Load_CECLM_general();
|
|
views = [0,0,0; 0,-30,0; 0,30,0; 0,-55,0; 0,55,0; 0,0,30; 0,0,-30; 0,-90,0; 0,90,0; 0,-70,40; 0,70,-40];
|
|
views = views * pi/180;
|
|
|
|
% As early termination weights were trained on part of menpo turn them off,
|
|
% to perform a clean cross-data experiment
|
|
early_term_params.weights_scale(:) = 1;
|
|
early_term_params.weights_add(:) = 0;
|
|
early_term_params.cutoffs(:) = -0.2;
|
|
|
|
% Use the multi-hypothesis model, as bounding box tells nothing about
|
|
% orientation
|
|
multi_view = true;
|
|
|
|
%% Setup recording
|
|
experiment.params = clmParams;
|
|
|
|
num_points = numel(pdm.M)/3;
|
|
|
|
shapes_all = cell(numel(images), 1);
|
|
labels_all = cell(numel(images), 1);
|
|
lhoods = zeros(numel(images),1);
|
|
all_lmark_lhoods = zeros(num_points, numel(images));
|
|
all_views_used = zeros(numel(images),1);
|
|
|
|
% Change if you want to visualize the outputs
|
|
verbose = false;
|
|
output_img = false;
|
|
|
|
if(output_img)
|
|
output_root = './ceclm_gen_out/';
|
|
if(~exist(output_root, 'dir'))
|
|
mkdir(output_root);
|
|
end
|
|
end
|
|
if(verbose)
|
|
f = figure;
|
|
end
|
|
%% Fitting the model to the provided images
|
|
tic
|
|
for i=1:numel(images)
|
|
image = imread(images(i).img);
|
|
image_orig = image;
|
|
|
|
if(size(image,3) == 3)
|
|
image = rgb2gray(image);
|
|
end
|
|
|
|
bbox = squeeze(detections(i,:));
|
|
|
|
% The actual work get's done here
|
|
[shape,~,~,lhood,lmark_lhood,view_used] =...
|
|
Fitting_from_bb_multi_hyp(image, [], bbox, pdm, patches, clmParams, views, early_term_params);
|
|
|
|
all_lmark_lhoods(:,i) = lmark_lhood;
|
|
all_views_used(i) = view_used;
|
|
|
|
shapes_all{i} = shape;
|
|
labels_all{i} = labels{i};
|
|
|
|
if(mod(i, 200)==0)
|
|
fprintf('%d done\n', i );
|
|
end
|
|
|
|
lhoods(i) = lhood;
|
|
|
|
if(output_img)
|
|
v_points = logical(patches(1).visibilities(view_used,:))';
|
|
DrawFaceOnImg(image_orig, shape, sprintf('%s/%s%d.jpg', output_root, 'fit', i), bbox, v_points);
|
|
end
|
|
|
|
if(verbose)
|
|
v_points = logical(patches(1).visibilities(view_used,:))';
|
|
DrawFaceOnFig(image_orig, shape, bbox, v_points);
|
|
end
|
|
|
|
end
|
|
toc
|
|
|
|
experiment.lhoods = lhoods;
|
|
experiment.shapes = shapes_all;
|
|
experiment.labels = labels_all;
|
|
experiment.all_lmark_lhoods = all_lmark_lhoods;
|
|
experiment.all_views_used = all_views_used;
|
|
|
|
%%
|
|
output_results = 'results/results_ceclm_cross-data.mat';
|
|
save(output_results, 'experiment');
|
|
|
|
end
|