add options to the -metrics arg

ivadomed · Jun 6, 2024 · f4b1f68 · f4b1f68
1 parent be048fb
commit f4b1f68
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/compute_metrics_reloaded.py b/compute_metrics_reloaded.py
@@ -72,8 +72,10 @@ def get_parser():
     parser.add_argument('-reference', required=True, type=str,
                         help='Path to the folder with nifti images of reference (ground truth) or path to a single '
                              'nifti image of reference (ground truth).')
-    parser.add_argument('-metrics', nargs='+', default=['dsc', 'fbeta', 'nsd', 'vol_diff', 'rel_vol_error'],
-                        required=False,
+    parser.add_argument('-metrics', nargs='+', required=False,
+                        default=['dsc', 'fbeta', 'nsd', 'vol_diff', 'rel_vol_error',
+                                 'lesion_ppv', 'lesion_sensitivity', 'lesion_f1_score'
+                                 'ref_count', 'pred_count', 'lcwa'],
                         help='List of metrics to compute. For details, '
                              'see: https://metricsreloaded.readthedocs.io/en/latest/reference/metrics/metrics.html. '
                              'Default: dsc, fbeta, nsd, vol_diff, rel_vol_error')