Commit 41b559ac authored by Alexander Kolesnikov's avatar Alexander Kolesnikov Committed by Ross Girshick
Browse files

fix the model path in the demo

parent ae903eac
...@@ -124,8 +124,9 @@ def parse_args(): ...@@ -124,8 +124,9 @@ def parse_args():
if __name__ == '__main__': if __name__ == '__main__':
args = parse_args() args = parse_args()
prototxt = os.path.join('models', NETS[args.demo_net][0], 'test.prototxt') prototxt = os.path.join(cfg.ROOT_DIR, 'models', NETS[args.demo_net][0],
caffemodel = os.path.join('data', 'fast_rcnn_models', 'test.prototxt')
caffemodel = os.path.join(cfg.ROOT_DIR, 'data', 'fast_rcnn_models',
NETS[args.demo_net][1]) NETS[args.demo_net][1])
if not os.path.isfile(caffemodel): if not os.path.isfile(caffemodel):
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment