Add multi-channel support for CUDA distance transforms

masadcv / FastGeodis

Fast Implementation of Generalised Geodesic Distance Transform for CPU (OpenMP) and GPU (CUDA)

https://fastgeodis.readthedocs.io

BSD 3-Clause "New" or "Revised" License

90 stars 14 forks source link

Closed masadcv closed 2 years ago

masadcv commented 2 years ago

This PR adds support for multi-channel input image for computing distance transforms for the CUDA kernels.

CPU code already does this.

Also fixed minor formatting and consistency issues.