Added default values to matcaffe_batch for testing

This commit is contained in:
Sergio Guadarrama 2014-03-31 15:55:33 -07:00
Родитель 38e3d7ff3a
Коммит 5e79c95eac
1 изменённых файлов: 7 добавлений и 1 удалений

Просмотреть файл

@ -16,14 +16,20 @@ function [scores,list_im] = matcaffe_batch(list_im, use_gpu)
% Or the equivalent based on where things are installed on your system
%
% Usage:
% scores = matcaffe_batch({'peppers.png','onion.png'}, 0);
% scores = matcaffe_batch({'peppers.png','onion.png'});
% scores = matcaffe_batch('list_images.txt', 1);
if nargin < 1
% For test purposes
list_im = {'peppers.png','onions.png'};
end
if ischar(list_im)
%Assume it is a file contaning the list of images
filename = list_im;
list_im = read_cell(filename);
end
% Adjust the batch size to match with imagenet_deploy.prototxt
batch_size = 10;
% Adjust dim to the output size of imagenet_deploy.prototxt
dim = 1000;
disp(list_im)
if mod(length(list_im),batch_size)