Total Segmentator on 5.2.2 "scikit-learn" distribution not found

thanks very much.
the output results is as follows:
Processing started
Writing input file to C:/Users/whu/AppData/Local/Temp/Slicer/__SlicerTemp__2023-04-25_19+22+09.770/total-segmentator-input.nii
Creating segmentations with TotalSegmentator AI…
Total Segmentator arguments: [‘-i’, ‘C:/Users/whu/AppData/Local/Temp/Slicer/__SlicerTemp__2023-04-25_19+22+09.770/total-segmentator-input.nii’, ‘-o’, ‘C:/Users/whu/AppData/Local/Temp/Slicer/__SlicerTemp__2023-04-25_19+22+09.770/segmentation’, ‘–ml’, ‘–task’, ‘total’, ‘–fast’]
Traceback (most recent call last):
File “C:\SlicerInstall\Slicer 5.3.0\lib\Python\Scripts\TotalSegmentator”, line 93, in
main()
File “C:\SlicerInstall\Slicer 5.3.0\lib\Python\Scripts\TotalSegmentator”, line 86, in main
totalsegmentator(args.input, args.output, args.ml, args.nr_thr_resamp, args.nr_thr_saving,
File “C:\SlicerInstall\Slicer 5.3.0\lib\Python\Lib\site-packages\totalsegmentator\python_api.py”, line 173, in totalsegmentator
seg = nnUNet_predict_image(input, output, task_id, model=model, folds=folds,
File “C:\SlicerInstall\Slicer 5.3.0\lib\Python\Lib\site-packages\totalsegmentator\nnunet.py”, line 255, in nnUNet_predict_image
nnUNet_predict(tmp_dir, tmp_dir, task_id, model, folds, trainer, tta)
File “C:\SlicerInstall\Slicer 5.3.0\lib\Python\Lib\site-packages\totalsegmentator\nnunet.py”, line 106, in nnUNet_predict
predict_from_folder(model_folder_name, dir_in, dir_out, folds, save_npz, num_threads_preprocessing,
File “C:\SlicerInstall\Slicer 5.3.0\lib\Python\Lib\site-packages\nnunet\inference\predict.py”, line 616, in predict_from_folder
shutil.copy(join(model, ‘plans.pkl’), output_folder)
File “C:\SlicerInstall\Slicer 5.3.0\lib\Python\Lib\shutil.py”, line 427, in copy
copyfile(src, dst, follow_symlinks=follow_symlinks)
File “C:\SlicerInstall\Slicer 5.3.0\lib\Python\Lib\shutil.py”, line 264, in copyfile
with open(src, ‘rb’) as fsrc:
FileNotFoundError: [Errno 2] No such file or directory: ‘C:\Users\whu\.totalsegmentator\nnunet\results\nnUNet\3d_fullres\Task256_TotalSegmentator_3mm_1139subj\nnUNetTrainerV2_ep8000_nomirror__nnUNetPlansv2.1\plans.pkl’
Exception ignored in: <totalsegmentator.libs.DummyFile object at 0x000001D0B7DD8A00>
AttributeError: ‘DummyFile’ object has no attribute ‘flush’

If you use this tool please cite: [2208.05868] TotalSegmentator: robust segmentation of 104 anatomical structures in CT images

Using ‘fast’ option: resampling to lower resolution (3mm)
Resampling…
Resampled in 0.47s
Predicting…

Thank you for posting the full output. This message confirms that the issue is that download of the model weights was incomplete. I would recommend to download the weights from Zenodo manually, using your web browser, and place it in the TotalSegmentator folder in your user folder, replacing the incomplete download that is already there.

1 Like

I solved it. thanks :smile: