search

fepegar / torchio

Medical imaging toolkit for deep learning
http://www.torchio.org
Apache License 2.0
2.02k stars 234 forks source link

Complete label sampler #948

Open themantalope opened 1 year ago

themantalope commented 1 year ago

🚀 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 commented 1 year ago

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