-
Notifications
You must be signed in to change notification settings - Fork 74
/
Copy pathsetup.m
29 lines (26 loc) · 892 Bytes
/
setup.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
function setup(varargin)
% SETUP Add the required search paths to MATLAB
run matconvnet/matlab/vl_setupnn ;
addpath matconvnet/examples ;
opts.useGpu = false ;
opts.verbose = false ;
opts.enableImReadJPEG = false ;
opts = vl_argparse(opts, varargin) ;
try
vl_nnconv(single(1),single(1),[]) ;
catch
warning('VL_NNCONV() does not seem to be compiled. Trying to compile it now.') ;
vl_compilenn('enableGpu', opts.useGpu, ...
'enableImReadJPEG', opts.enableImReadJPEG, ...
'verbose', opts.verbose) ;
end
if opts.useGpu
try
vl_nnconv(gpuArray(single(1)),gpuArray(single(1)),[]) ;
catch
vl_compilenn('enableGpu', opts.useGpu, ...
'enableImReadJPEG', opts.enableImReadJPEG, ...
'verbose', opts.verbose) ;
warning('GPU support does not seem to be compiled in MatConvNet. Trying to compile it now') ;
end
end