From 0f9535e6d9bc4279eafb8e3eb91f47773498adb8 Mon Sep 17 00:00:00 2001 From: Jamie Date: Wed, 20 Mar 2019 21:45:24 +0800 Subject: [PATCH 1/2] Update infer.py --- infer.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/infer.py b/infer.py index ce76955..9df2bff 100755 --- a/infer.py +++ b/infer.py @@ -67,6 +67,8 @@ def main(): parser.add_argument('--training-data', default='pascal-voc/training-data.pkl', help='Information about parameters used for training') + parser.add_argument('--data-source', default='pascal_voc', + help='data source') parser.add_argument('--output-dir', default='test-output', help='directory for the resulting images') parser.add_argument('--annotate', type=str2bool, default='False', From 12c69d51fbc33645b276671b8ab7053227aaf7f7 Mon Sep 17 00:00:00 2001 From: Jamie Date: Wed, 20 Mar 2019 21:45:50 +0800 Subject: [PATCH 2/2] Update source_pascal_voc.py --- source_pascal_voc.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/source_pascal_voc.py b/source_pascal_voc.py index f5a6809..d7ada07 100644 --- a/source_pascal_voc.py +++ b/source_pascal_voc.py @@ -192,10 +192,10 @@ def load_test_data(self, data_dir): Load the test data :param data_dir: the directory where the dataset's file are stored """ - root = data_dir + '/test/VOCdevkit/VOC2012' + root = data_dir + '/test/VOCdevkit/VOC2007' annot = self.__build_annotation_list(root, 'test') self.test_samples = self.__build_sample_list(root, annot, - 'test_VOC2012') + 'test_VOC2007') if len(self.test_samples) == 0: raise RuntimeError('No testing samples found in ' + data_dir)