【PyTorch】使用 MaxPool 实现图像的膨胀和腐蚀

2023-05-16

PyTorch使用 MaxPool 实现图像的膨胀和腐蚀:

max_pool = nn.MaxPool2d(kernel_size=5, stride=1, padding=2)  # 可调整 kernel_size 和 padding
# 假设要处理的张量为 tensor
# ===== 膨胀 =====
tensor_dilate = max_pool(tensor)
# ===== 腐蚀 =====
tensor_erode = -max_pool(-tensor)

————————————————
来源:https://blog.csdn.net/qq_39546227/article/details/122473559

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

【PyTorch】使用 MaxPool 实现图像的膨胀和腐蚀 的相关文章

随机推荐