TorchMorph 发布:PyTorch 生态的 CUDA 加速形态学库

Kai Zhao 发布 TorchMorph,一个基于 PyTorch 的形态学变换扩展库,提供跨高达八维的 GPU 加速形态学和距离变换算子,并采用与 SciPy 兼容的 API。该库针对 GPU 训练循环优化,通过融合 CUDA 内核直接在 CUDA 张量上操作,旨在显著加速计算图像处理任务。

2026-08-26 ~ 2026-08-26 · 2 条相关