TorchMorph：CUDA 加速的形态学变换库
TorchMorph: CUDA-accelerated Morphological Transforms
阅读原文
本站未展示全文，请前往来源网站阅读。
AI 导读
TorchMorph 是一个轻量级 PyTorch 扩展，用融合 CUDA kernel 直接在 (B, C, Spatial...) 张量上实现形态学变换，支持最多八个空间维度、共 22 个公开算子，涵盖二值形态学、灰度形态学、精确与近似距离变换以及熵正则最优传输。
来源：Hugging Face Daily Papers · arxiv.org