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

kornia_foss · x · 2026-08-26

Kai Zhao 发布了 TorchMorph,这是一个基于 PyTorch 的新形态学变换库。它针对 GPU 训练循环进行了优化,通过融合 CUDA 内核直接在 CUDA 张量上操作。与 CPU 版本的 SciPy 相比,批处理执行在灰度形态学上吞吐量最高提升 1100 倍,精确欧几里得距离变换最高提升 350 倍。API 设计与 SciPy 高度兼容,便于现有管线迁移。

所属事件:TorchMorph 发布:PyTorch 生态的 CUDA 加速形态学库(2 条相关)→

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →