Complete label sampler

[themantalope]

🚀 Feature

Sampler which is guaranteed to include all voxels with a certain label value.

Motivation

I am doing tumor localization/segmentation, and I need to sample patches of the images due to memory and spatial resolution constraints. I need to make sure that the tumors are within the sampled patches with a certain number of pixels/voxels between the edge of the tumor and the edge of the patch.

Pitch

Add a class to perform complete label sampling.

Alternatives

I've already rolled my own code to do this, but it might be useful for other folks who are working on similar things. Would be happy to contribute or figure out a way to add a class that is in line with the torchio framework.

[fepegar]

Hi, @themantalope. Thanks for the pitch. I'm not sure what you mean by "complete label sampling". Could you please elaborate?