From 415ada6d2bae8af908e5ded4945f318ba1c6e21a Mon Sep 17 00:00:00 2001 From: Christian Forster Date: Thu, 10 Apr 2014 22:49:24 +0200 Subject: [PATCH] copy dataset parameters to the trace folder --- svo_analysis/scripts/benchmark.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/svo_analysis/scripts/benchmark.py b/svo_analysis/scripts/benchmark.py index 9611350..4b11148 100755 --- a/svo_analysis/scripts/benchmark.py +++ b/svo_analysis/scripts/benchmark.py @@ -13,6 +13,7 @@ import time import vikit_py.cpu_info as cpu_info import vikit_py.ros_node as ros_node import evaluate +import shutil def run_experiment(dataset, params): # load dataset parameters @@ -29,11 +30,14 @@ def run_experiment(dataset, params): # combine all parameters params = dict(params.items() + algo_params.items() + dataset_params['rig']['cam0'].items()) - - # dump experiment params to file: - params_dump_file = params['trace_dir']+'/params.yaml' + if 'dataset_is_blender' in dataset_params: + params['dataset_is_blender'] = True + + # dump experiment params to file and copy the other parameter files: + params_dump_file = os.path.join(params['trace_dir'],'params.yaml') with open(params_dump_file,'w') as outfile: outfile.write(yaml.dump(params, default_flow_style=False)) + shutil.copyfile(dataset_params_file, os.path.join(params['trace_dir'], 'dataset_params.yaml')) # execute ros node node = ros_node.RosNode(args.version, args.executable)