liver_vessels subtask invokes model download everytime

wasserth / TotalSegmentator

Tool for robust segmentation of >100 important anatomical structures in CT and MR images

Apache License 2.0

1.45k stars 245 forks source link

liver_vessels subtask invokes model download everytime #371

Open ASHISRAVINDRAN opened 3 days ago

ASHISRAVINDRAN commented 3 days ago

Hi @wasserth I am from the MITK development team working to update TotalSegmentator to v2.4 for our upcoming release. I am testing out inferencing all subtasks as part of sanity test. I see that subtask liver_vessels invokes model download from internet on every run. Its weights are not ultimately saved in the .totalsegmentator folder found in the $HOME directory after the first run. It would be great if you can check this.

Thanks in advance

wasserth commented 2 days ago

Unfortunately I can not reproduce this issue. For me the weights are properly stored in the default directory ~/.totalsegmentator/nnunet/results. Which operating system are you using?

ASHISRAVINDRAN commented 2 days ago

I can reproduce on Windows 10 and Ubuntu 22.04. When I run Totalsegmentator -i .\example_ct.nii.gz -o .\outDir\ -ta liver_vessels I can see a tmp_download_file.zip file in the ~/.totalsegmentator/nnunet/results folder. After inferencing, its gone. Its not persisted or renamed to something in the lines of "DatasetXXX_subtask_partX..."