refactor: Minor cleanup

Author: Karol-G

README.md

@@ -1,8 +1,5 @@
 # ScribbleBench
 
-[![License Apache Software License 2.0](https://img.shields.io/pypi/l/ScribbleBench.svg?color=green)](https://github.com/Karol-G/ScribbleBench/raw/main/LICENSE)
-[![Python Version](https://img.shields.io/pypi/pyversions/ScribbleBench.svg?color=green)](https://python.org)
-
 **ScribbleBench** is a comprehensive benchmark for evaluating the generalization capabilities of 3D scribble-supervised medical image segmentation methods. It spans seven diverse datasets across multiple anatomies and modalities and provides realistic, automatically generated scribble annotations.
 
 This repository provides:

scribblebench/evaluation.py

@@ -9,8 +9,8 @@
 def evaluate(gt_dir, pred_dir, num_classes, num_processes=None):    
     gt_dir = Path(gt_dir)
     pred_dir = Path(pred_dir)
-    names_gt = [path.name[:-7] for path in Path(gt_dir).rglob("*.nii.gz")]
-    names_pred = [path.name[:-7] for path in Path(pred_dir).rglob("*.nii.gz")]
+    names_gt = [path.name[:-7] for path in gt_dir.rglob("*.nii.gz")]
+    names_pred = [path.name[:-7] for path in pred_dir.rglob("*.nii.gz")]
 
     if set(names_gt) != set(names_pred):
         raise RuntimeError(f"The set of GT segmentations is different to the set of predictions. Do you have missing predictions?")