diff --git a/compute_metrics/compute_metrics_reloaded.py b/compute_metrics/compute_metrics_reloaded.py index 9c83575..ce7050b 100644 --- a/compute_metrics/compute_metrics_reloaded.py +++ b/compute_metrics/compute_metrics_reloaded.py @@ -45,6 +45,16 @@ from MetricsReloaded.metrics.pairwise_measures import BinaryPairwiseMeasures as BPM +METRICS_TO_NAME = { + 'dsc': 'Dice similarity coefficient (DSC)', + 'hd': 'Hausdorff distance', + 'fbeta': 'FBeta score', + 'nsd': 'Normalized surface distance (NSD)', + 'vol_diff': 'Volume difference', + 'rel_vol_diff': 'Relative volume error (RVE)', +} + + def get_parser(): # parse command line arguments parser = argparse.ArgumentParser(description='Compute MetricsReloaded metrics for segmentation tasks.') @@ -223,6 +233,9 @@ def main(): # Convert JSON data to pandas DataFrame df = build_output_dataframe(output_list) + # Rename columns + df.rename(columns={metric: METRICS_TO_NAME[metric] for metric in METRICS_TO_NAME}, inplace=True) + # save as CSV fname_output_csv = os.path.abspath(args.output) df.to_csv(fname_output_csv, index=False)