fastgeodis (pypi): Fast Implementation of Generalised Geodesic Distance Transform for CPU (OpenMP) and GPU (CUDA)
Keywords: deep-learning|gpu|pytorch