basalt/scripts/eval_full/gen_results.py

#!/usr/bin/env python

import os
import sys


datasets = ['MH_01_easy', 'MH_02_easy', 'MH_03_medium', 'MH_04_difficult',
'MH_05_difficult', 'V1_01_easy', 'V1_02_medium',
'V1_03_difficult', 'V2_01_easy', 'V2_02_medium']


# Other results.
results_vio = []
results_mapping = []

out_dir = sys.argv[1]


for key in datasets:
    fname = out_dir + '/vio_' + key
    if os.path.isfile(fname): 
        with open(fname, 'r') as f:
            res = round(float(f.read()), 3)
        results_vio.append(float(res))
    else:
        results_vio.append(float('Inf'))
    
    fname = out_dir + '/mapper_' + key
    if os.path.isfile(fname): 
        with open(fname, 'r') as f:
            res = round(float(f.read()), 3)
        results_mapping.append(float(res))
    else:
        results_mapping.append(float('Inf'))

row_format ="{:>17}" * (len(datasets))
print row_format.format(*datasets)
print row_format.format(*results_vio)
print row_format.format(*results_mapping)