taigw / brats17

Brain tumor segmentation for MICCAI 2017 BraTS challenge
BSD 3-Clause "New" or "Revised" License
324 stars 130 forks source link

How to enable NiftyNet's balanced window sampler? #37

Open pranavkantgaur opened 5 years ago

pranavkantgaur commented 5 years ago

Hi,

I have a highly imbalanced data-distributon(3-4% of total voxels represent the object of interest) for my problem. I would like to know how to enable Niftynet's balanced window sampler here? Do I need to modify the get_subimage_batch() method?

Further, would you please suggest other features of your code which can be used to address the class-imbalance problem. I am using ROI patches for training with improved results and planning to use the tversky loss function in future.