conv_op=nn.Conv2d(1,1,3, bias=False) # 定义sobel算子参数 sobel_kernel=np.array([[-1,-1,-1], [-1,8,-1], [-1,-1,-1]], dtype='float32') #将sobel算子转换为适配卷积操作的卷积核 sobel_kernel=sobel_kernel.reshape((1,1,3,3)) # 给卷积操作的卷积核赋值 conv_op.weight.data=...
www.jb51.net/article/1782...htm 2024-5-27