funkelab / gunpowder

A library to facilitate machine learning on multi-dimensional images.
https://funkelab.github.io/gunpowder/
MIT License
78 stars 56 forks source link

Add support for providing random shift and random provider to batch #173

Closed pattonw closed 2 years ago

pattonw commented 2 years ago

Pre-Merge Checklist:

You can now provide the RandomLocation node an array key which it will populate with the random shift chosen for each batch. You can now provide the RandomProvider node an array key which it will populate with the index of the random provider chosen for each batch.

TODO: same thing for other random nodes such as elastic augment, shift augment etc.