fix demo.py

xhzdeng · Apr 18, 2018 · a5aef0f · a5aef0f
1 parent c646be8
commit a5aef0f
Show file tree

Hide file tree

Showing 5 changed files with 4 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -60,7 +60,8 @@ This code is based on [Caffe](https://github.com/BVLC/caffe) and [py-faster-rcnn
     output/<experiment directory>/<dataset name>/<network snapshot name>/
     ```
 
-7. Demo
+
+### Demo
 	```
 	cd $CRPN_ROOT
 	./tools/demo.py --net [NET] --model [MODEL]

diff --git a/data/demo/img_10.jpg b/data/demo/img_10.jpg
diff --git a/data/demo/img_14.jpg b/data/demo/img_14.jpg
diff --git a/data/demo/img_45.jpg b/data/demo/img_45.jpg
diff --git a/tools/demo.py b/tools/demo.py
@@ -153,6 +153,7 @@ def parse_args():
     #
     if args.demo_net is None:
         prototxt = os.path.join(cfg.MODELS_DIR, NETS[args.demo_net][0], 'faster_rcnn_end2end', 'test.prototxt')
+        cfg_file = None
     else:
         prototxt = os.path.join('./models', args.demo_net, 'test.pt')
         cfg_file = os.path.join('./models', args.demo_net, 'config.yml')
@@ -184,8 +185,7 @@ def parse_args():
     # for i in xrange(2):
     #     _, _= im_detect(net, im)
 
-    im_names = ['img_1.jpg', 'img_2.jpg', 'img_3.jpg',
-                'img_4.jpg', 'img_5.jpg']
+    im_names = ['img_10.jpg', 'img_14.jpg', 'img_45.jpg']
     for im_name in im_names:
         print '~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~'
         print 'Demo for data/demo/{}'.format(im_name)