diff --git a/README.md b/README.md index 64e024e..f98a22c 100755 --- 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//// ``` -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 new file mode 100644 index 0000000..c005551 Binary files /dev/null and b/data/demo/img_10.jpg differ diff --git a/data/demo/img_14.jpg b/data/demo/img_14.jpg new file mode 100644 index 0000000..688f6a0 Binary files /dev/null and b/data/demo/img_14.jpg differ diff --git a/data/demo/img_45.jpg b/data/demo/img_45.jpg new file mode 100644 index 0000000..a8e5a1b Binary files /dev/null and b/data/demo/img_45.jpg differ diff --git a/tools/demo.py b/tools/demo.py index 98930c2..e463c8f 100755 --- 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)