fix the model path in the demo

41b559ac · Alexander Kolesnikov · Ross Girshick · ae903eac · 41b559ac
隐藏空白更改
内联并排

Showing with 3 addition and 2 deletion

tools/demo.py tools/demo.py +3 -2

未找到文件。
--- 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):