fixed a bug in generate_dataset_json

MIC-DKFZ · May 7, 2021 · 058b695 · 058b695
1 parent 6b08055
commit 058b695
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nnunet/dataset_conversion/utils.py b/nnunet/dataset_conversion/utils.py
@@ -62,7 +62,7 @@ def generate_dataset_json(output_file: str, imagesTr_dir: str, imagesTs_dir: str
     json_dict['labels'] = {str(i): labels[i] for i in labels.keys()}
 
     json_dict['numTraining'] = len(train_identifiers)
-    json_dict['numTest'] = []
+    json_dict['numTest'] = len(test_identifiers)
     json_dict['training'] = [
         {'image': "./imagesTr/%s.nii.gz" % i, "label": "./labelsTr/%s.nii.gz" % i} for i
         in