search

wasserth / TotalSegmentator

Tool for robust segmentation of >100 important anatomical structures in CT and MR images
Apache License 2.0
1.53k stars 253 forks source link

Question regarding the --roi_subset accuracy. #367

Closed maksymfritsak closed 1 month ago

maksymfritsak commented 1 month ago

In your README, it is specified that --roi_subset might be less accurate, especially for small classes (e.g., prostate).

Is the accuracy lower because the model makes predictions on other derived structures, and when predicting only the prostate, the model does not have other structures to use as "support" for the segmentation?

wasserth commented 1 month ago

It can be lower, because first a lower resolution model searches the organ and then the image is cropped to the organs and then the highres model is run on this smaller crop. The highres model has the same accuracy. But it can happen that the lower resolution model does not find the organ properly and then the crop is wrong.