search

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_affinities: dtype, channel dim #125

Closed abred closed 4 years ago

abred commented 4 years ago

add arg to set dtype

check if array has unwanted channel dim before computing affinities

pattonw commented 4 years ago

I like the affinities dtype attribute, I think that can be merged without question, but the axis squeeze should be removed or moved to a different pull request.

abred commented 4 years ago

I split it in two (#133, #134 )