BraTS-Toolkit icon indicating copy to clipboard operation
BraTS-Toolkit copied to clipboard

Minimal segmentation requirements

Open BegoneGIT opened this issue 7 months ago • 1 comments

Hi! I'd like to ask what are reccomended specs to run segmentor on? I've i5-9300HF, 8GB GeForce GTX 1650, so not an amazing spec.

I ask this since isen-20 hangs after ~20 minutes (docker image CPU and RAM usage goes to 0, but virtual machine itself still hogs on those resorces somehow). To free up the memory I have to quit Docker altogether. I tried deleting and recreating this docker image, but it doesn't seem to help. Last thing printed in console is Executing: docker run --rm --gpus device=0 -v \BraTS-Toolkit\brats_toolkit\tmpps3eqv8s:/app/data/ brats/isen-20 python runner.py.

I think it's memory issue because sanet0-20 returned CUDA out of memory error.

I was able to run examples and they did return proper segmentation masks in nii.gz.

Thinking about possible solution I came up with two things. For first maybe I could 'reduce' the data somehow so models have more memory to work on. Second is, maybe there are smaller (or pruned) models I could use.

BegoneGIT avatar Nov 28 '23 19:11 BegoneGIT