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