From c460e3bf853085c076f77772ffef4ef787ee27e4 Mon Sep 17 00:00:00 2001 From: Christian Forster Date: Tue, 8 Apr 2014 20:13:57 +0200 Subject: [PATCH] adapt for new datasets --- svo_analysis/scripts/benchmark.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/svo_analysis/scripts/benchmark.py b/svo_analysis/scripts/benchmark.py index 19688ce..3be6916 100755 --- a/svo_analysis/scripts/benchmark.py +++ b/svo_analysis/scripts/benchmark.py @@ -38,7 +38,7 @@ if __name__=="__main__": expParams['experiment_name'] = expParams['time']+'_'+args.version+'_' + args.experiment_file # load dataset parameters - expParams['dataset_directory'] = os.path.join(rospkg.RosPack().get_path('Datasets'), expParams['dataset']) + expParams['dataset_directory'] = os.path.join(os.environ['SVO_DATASET_DIR'], expParams['dataset']) if not os.path.exists(expParams['dataset_directory']): raise Exception("Provided dataset folder does not exist.") datasetParamsFile = os.path.join(expParams['dataset_directory'], 'dataset_params.yaml') @@ -50,10 +50,12 @@ if __name__=="__main__": algoParams = yaml.load(open(algoParamsFile,'r')) # combine all parameters - params = dict(expParams.items() + datasetParams.items() + algoParams.items()) + params = dict(expParams.items() + datasetParams.items() + datasetParams['rig']['cam0'].items() + algoParams.items()) params['trace_name'] = 'trace' params['trace_dir'] = os.path.join(rospkg.RosPack().get_path('svo_analysis'), 'results', params['experiment_name']) + + # create folder for tracing if not os.path.exists(params['trace_dir']):