Skip to content

Commit

Permalink
debug matlab wrapper
Browse files Browse the repository at this point in the history
  • Loading branch information
kevinlin311tw committed Jan 8, 2016
1 parent 0274957 commit 35bac78
Showing 1 changed file with 1 addition and 9 deletions.
10 changes: 1 addition & 9 deletions matlab/feat_batch.m
Original file line number Diff line number Diff line change
Expand Up @@ -4,14 +4,6 @@
% By default use CPU
use_gpu = 0;
end
if nargin < 2 || isempty(net_model)
% By default use imagenet_deploy
net_model = '/home/titan/hfyang/hashing/training/NUS-WIDE/NUS-KevinNet-48/KevinNet_NUS_48_deploy.prototxt';
end
if nargin < 3 || isempty(net_weights)
% By default use caffe reference model
net_weights = '/home/titan/hfyang/hashing/training/NUS-WIDE/NUS-KevinNet-48/models-20150720/KevinNet_NUS_48_iter_50000';
end
if ischar(list_im)
%Assume it is a file contaning the list of images
filename = list_im;
Expand Down Expand Up @@ -41,7 +33,7 @@
net = caffe.Net(net_model, net_weights, phase);

% load mean file
d = load('/home/iis/adsc/caffe-new-cbd-udnn/matlab/+caffe/imagenet/ilsvrc_2012_mean.mat');
d = load('./matlab/+caffe/imagenet/ilsvrc_2012_mean.mat');
mean_data = d.mean_data;

batch_size = 10;
Expand Down

0 comments on commit 35bac78

Please sign in to comment.