public/vio
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

copy dataset parameters to the trace folder

Browse Source
This commit is contained in:
Christian Forster
2014-04-10 22:49:24 +02:00
parent fedd9f250e
commit 415ada6d2b
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
svo_analysis/scripts/benchmark.py
+7 -3
View File
@@ -13,6 +13,7 @@ import time
import vikit_py.cpu_info as cpu_info import vikit_py.cpu_info as cpu_info
import vikit_py.ros_node as ros_node import vikit_py.ros_node as ros_node
import evaluate import evaluate
import shutil
def run_experiment(dataset, params): def run_experiment(dataset, params):
# load dataset parameters # load dataset parameters
@@ -29,11 +30,14 @@ def run_experiment(dataset, params):
# combine all parameters # combine all parameters
params = dict(params.items() + algo_params.items() + dataset_params['rig']['cam0'].items()) params = dict(params.items() + algo_params.items() + dataset_params['rig']['cam0'].items())
if 'dataset_is_blender' in dataset_params:
# dump experiment params to file: params['dataset_is_blender'] = True
params_dump_file = params['trace_dir']+'/params.yaml'
# 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: with open(params_dump_file,'w') as outfile:
outfile.write(yaml.dump(params, default_flow_style=False)) 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 # execute ros node
node = ros_node.RosNode(args.version, args.executable) node = ros_node.RosNode(args.version, args.executable)