This commit is contained in:
Evan Shelhamer 2014-03-13 20:13:46 -07:00
Родитель 04d6595d3c
Коммит 0816a94803
2 изменённых файлов: 5 добавлений и 5 удалений

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

@ -1,6 +1,6 @@
function scores = matcaffe_demo(im, use_gpu)
% scores = matcaffe_demo(im, use_gpu)
%
%
% Demo of the matlab wrapper using the ILSVRC network.
%
% input
@ -20,9 +20,9 @@ function scores = matcaffe_demo(im, use_gpu)
% scores = matcaffe_demo(im, 1);
% [score, class] = max(scores);
model_def_file = '../../examples/imagenet_deploy.prototxt';
model_def_file = '../../models/imagenet.prototxt';
% NOTE: you'll have to get the pre-trained ILSVRC network
model_file = '../../examples/alexnet_train_iter_470000';
model_file = '../../models/caffe_reference_imagenet_model';
% init caffe network (spews logging info)
caffe('init', model_def_file, model_file);

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

@ -371,12 +371,12 @@ if __name__ == "__main__":
# Optional arguments.
parser.add_argument(
"--model_def",
default="examples/imagenet_deploy.prototxt",
default="../../../models/imagenet.prototxt",
help="Model definition file."
)
parser.add_argument(
"--pretrained_model",
default="examples/caffe_reference_imagenet_model",
default="../../../models/caffe_reference_imagenet_model",
help="Trained model weights file."
)
parser.add_argument(