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

_Suraimu_

3952人浏览 · 2022-01-13 14:43:51

_Suraimu_ · 2022-01-13 14:43:51 发布

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)