Skip to content

Commit

Permalink
Merge pull request #239 from fohofmann/master
Browse files Browse the repository at this point in the history
quiet to "predicting part x"
  • Loading branch information
wasserth authored Nov 27, 2023
2 parents adf8140 + e410654 commit 127f243
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion totalsegmentator/nnunet.py
Original file line number Diff line number Diff line change
Expand Up @@ -368,7 +368,7 @@ def nnUNet_predict_image(file_in, file_out, task_id, model="3d_fullres", folds=N
seg_combined[img_part] = np.zeros(img_shape, dtype=np.uint8)
# Run several tasks and combine results into one segmentation
for idx, tid in enumerate(task_id):
print(f"Predicting part {idx+1} of {len(task_id)} ...")
if not quiet: print(f"Predicting part {idx+1} of {len(task_id)} ...")
with nostdout(verbose):
# nnUNet_predict(tmp_dir, tmp_dir, tid, model, folds, trainer, tta,
# nr_threads_resampling, nr_threads_saving)
Expand Down

0 comments on commit 127f243

Please sign in to comment.