Avoid double counting channels in elastic augment reshape

funkelab / gunpowder

A library to facilitate machine learning on multi-dimensional images.

https://funkelab.github.io/gunpowder/

MIT License

79 stars 56 forks source link

Avoid double counting channels in elastic augment reshape #69

Closed cmalinmayor closed 5 years ago

cmalinmayor commented 5 years ago

Recently I got the error that elastic augment was trying to reshape my data into (7, 7, z, y, x) - it seemed to be double counting my time dimension. I tracked the source of the error back to this commit. I'm not sure what the original goal was exactly, but I took my best guess to make it work for me.