调用 opencv 函数的基本步骤如下：先把 pytorch 的 tensor 转到 cpu 上，然后转换成 numpy，再调整到 ...
Pytorch——基于max pooling实现图像腐蚀和膨胀 代码 可以通过调节kernel_size实现腐蚀和膨胀的大小。 import cv2 import torch import torch.nn.functional as F import numpy as np def dilate(bin_img, ksize=5): # 腐蚀 src_size = bin_img.numpy().shape pad = (ksize - 1) // 2 #...
使用MaxPool2d一行代码就搞定了。可以参考pytorch 20 基于pytorch实现腐蚀膨胀