fix wrapper example paths

caiom · Mar 14, 2014 · 0816a94 · 0816a94
1 parent 04d6595
commit 0816a94
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/matlab/caffe/matcaffe_demo.m b/matlab/caffe/matcaffe_demo.m
@@ -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 @@
 %  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);

diff --git a/python/caffe/detection/detector.py b/python/caffe/detection/detector.py
@@ -371,12 +371,12 @@ def config(model_def, pretrained_model, gpu, image_dim, image_mean_file):
   # 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(