TotalSegmentator results computed in fast mode are rough

Hi Everyone. Thank you Andras Lasso for this beautiful work.
I have a problem with total segmentator, please see the screenshot below:


Why the model is so rough? I have a GPU of 4 GB, maybe I have to use my CPU and wait one hour to have better outcomes? Or is it another problem?
Thank you in advance,
Francesco

Yes, exactly. A 4GB GPU is a solid device for visualization but does not meet hardware requirements for most deep learning applications.

If you want to get a GPU that can run most image segmentation models then I would recommend to get a one with at least 12GB RAM (such as an RTX 3060, it only costs around $600). If you want to a GPU that you can use for a few more years or you want to train your own models then you may consider getting one with 24GB memory (such as an RTX 3090, for about $2500).

If you don’t want to spend any money on new hardware then you can stay on the CPU and wait out the 10-30x longer computation times (40 minutes on CPU instead of 2 minutes on GPU). These long computation times are not tolerable if you want to use interactive segmentation methods or if you need to train models, but if you are using well-trained, robust, fully-automatic models then waiting an hour for a segmentation result should be fine in most cases. It would be still much faster than manual segmentation, while requiring zero manual effort.

2 Likes

Just for the records, I am just working on a Windows 10 with GTX 970 with 4 GB video RAM and can totalsegment a rather thin-sliced lung CT ( ~ 100 MB nrrd) with lobe smoothing in 80.3 s

Hi Andras. is there anywhere in a document that isn’t just a forum post where minimum requirements are listed. i’m approaching our IT department separate to our Radiation Therapy department to ask for some hardware to do this work and they want to see what the minimum working specs would be.

don’t think they’ll be happy with a forum post reply.

They can directly talk to package developers on this forum, so they should be happy, but you can find some specific data on GPU memory requirements here: GitHub - wasserth/TotalSegmentator: Tool for robust segmentation of 104 important anatomical structures in CT images

Based on specific images the GPU memory requirement for full-resolution segmentation is about 12GB, so if you want to be safe I then it could make sense to use a GPU with 16GB memory.