From 0c624c25eeb712eea4a7b909d68bffe11a300ac2 Mon Sep 17 00:00:00 2001
From: Ross Girshick <ross.girshick@gmail.com>
Date: Fri, 17 Apr 2015 11:31:51 -0700
Subject: [PATCH] option to run in 'competition mode' for saving PASCAL eval
server results
---
tools/test_net.py | 8 ++++++++
1 file changed, 8 insertions(+)
diff --git a/tools/test_net.py b/tools/test_net.py
index 71a866d..3c8dc24 100755
--- a/tools/test_net.py
+++ b/tools/test_net.py
@@ -41,6 +41,8 @@ def parse_args():
parser.add_argument('--imdb', dest='imdb_name',
help='dataset to test',
default='voc_2007_test', type=str)
+ parser.add_argument('--comp', dest='comp_mode', help='competition mode',
+ default=False, type=bool)
if len(sys.argv) == 1:
parser.print_help()
@@ -72,4 +74,10 @@ if __name__ == '__main__':
net.name = os.path.splitext(os.path.basename(args.caffemodel))[0]
imdb = get_imdb(args.imdb_name)
+ if args.comp_mode:
+ if 'use_salt' in imdb.config:
+ imdb.config['use_salt'] = False
+ if 'cleanup' in imdb.config:
+ imdb.config['cleanup'] = False
+
fast_rcnn_test.test_net(net, imdb)
--
GitLab