fix the model path in the demo

danilons · May 6, 2015 · 41b559a · 41b559a
1 parent ae903ea
commit 41b559a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tools/demo.py b/tools/demo.py
@@ -124,8 +124,9 @@ def parse_args():
 if __name__ == '__main__':
     args = parse_args()
 
-    prototxt = os.path.join('models', NETS[args.demo_net][0], 'test.prototxt')
-    caffemodel = os.path.join('data', 'fast_rcnn_models',
+    prototxt = os.path.join(cfg.ROOT_DIR, 'models', NETS[args.demo_net][0],
+                            'test.prototxt')
+    caffemodel = os.path.join(cfg.ROOT_DIR, 'data', 'fast_rcnn_models',
                               NETS[args.demo_net][1])
 
     if not os.path.isfile(caffemodel):