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