Morphological Transforms

TorchMorph 为形态学变换带来 GPU 加速

TorchMorph 是一个新的 PyTorch 扩展，它将形态学变换带到 GPU 上，解决了现有仅 CPU 实现（如 SciPy）的局限性。它提供了 22 个 CUDA 加速算子，用于各种形态学类型和距离变换，支持多达八个空间维度。这带来了显著的性能提升，TorchMorph 在灰度形态学方面的吞吐量比 SciPy 高出 1100 倍，在距离变换方面高出 350 倍，同时保持了高精度。