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

fix the model path in the demo

parent ae903eac
......@@ -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):
......
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