forked from farrajota/caltech-extract-data-toolbox
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathextract_files.m
79 lines (74 loc) · 2.59 KB
/
extract_files.m
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
function extract_files( varargin )
%% Extracts all image and annotations files into a folder
%
% If an input folder is specified, this script will extract all images from
% all files (every frame from the dataset) and organizes it into
% corresponding folders inside a folder named 'extracted_files'. If no
% input path is specified, the folder will be created in the current
% directory.
%% Add eval code paths
addpath(genpath('./'));
%% Initializations
% function's inputs
if (nargin>=1),
if ~isempty(varargin{1}),
dataset_path = varargin{1};
else dataset_path = pwd; end
else dataset_path = '../../'; end
if (nargin >= 2),
if ~isempty(varargin{2}),
dataset_name = varargin{2};
else dataset_name = 'usa'; end
else dataset_name = 'usa'; end
if (nargin >= 3),
if ~isempty(varargin{3}),
store_name = varargin{3};
else store_name = '/extracted_data'; end
else store_name = '/extracted_data'; end
% create folders
if(~exist(strcat(dataset_path, store_name),'dir')), mkdir(strcat(dataset_path, store_name)); end
%% Extract image files
switch dataset_name
case 'usa' % Caltech Pedestrian Datasets (all)
% Train
fprintf('Unpacking train set data...');
dbInfo('usatrain', dataset_path);
dbExtract(dataset_path, strcat(dataset_path, store_name),1,1);
fprintf('\nDone.');
% Test
fprintf('\nUnpacking test set data...');
dbInfo('usatest', dataset_path);
dbExtract(dataset_path, strcat(dataset_path, store_name),1,1);
fprintf('\nDone.');
case 'inria' % INRIA dataset
% train
fprintf('Unpacking train set data...');
dbInfo('inriatrain', dataset_path);
dbExtract(dataset_path, strcat(dataset_path, store_name),1,1);
fprintf('\nDone.');
% Test
fprintf('\nUnpacking test set data...');
dbInfo('inriatest', dataset_path);
dbExtract(dataset_path, strcat(dataset_path, store_name),1,1);
fprintf('\nDone.');
case 'tudbrussels' % TUD-Brussels dataset
% test
fprintf('Unpacking test set data...');
dbInfo('tudbrussels', dataset_path);
dbExtract(dataset_path, strcat(dataset_path, store_name),1,1);
fprintf('\nDone.');
case 'eth' % ETH dataset
% test
fprintf('Unpacking test set data...');
dbInfo('eth', dataset_path);
dbExtract(dataset_path, strcat(dataset_path, store_name),1,1);
fprintf('\nDone.');
case 'daimler' % Daimler dataset
% test
fprintf('Unpacking test set data...');
dbInfo('daimler', dataset_path);
dbExtract(dataset_path, strcat(dataset_path, store_name),1,1);
fprintf('\nDone.');
otherwise, error('unknown data type: %s', dataset_name);
end
end