当前位置：网站首页>Pytorch uses maxpool to realize image expansion and corrosion

Pytorch uses maxpool to realize image expansion and corrosion

2022-07-28 05:38:00 【A little knowledge, a hundred Xiaosheng】

max_pool = nn.MaxPool2d(kernel_size=5, stride=1, padding=2)  #  Adjustable  kernel_size  and  padding
#  Suppose the tensor to be processed is  tensor
# =====  inflation  =====
tensor_dilate = max_pool(tensor)
# =====  corrosion  =====
tensor_erode = -max_pool(-tensor)