拉普拉斯边界提取算子pytorch实现

天天都在摸鱼的乐乐

于 2023-08-20 19:12:05 发布

阅读量611

点赞数 2

文章标签： pytorch 人工智能 python

本文链接：https://blog.csdn.net/qq_46401672/article/details/132394741

版权

import torch
import torch.nn as nn
class Laplace(nn.Module):
    def __init__(self):
        super(Laplace,self).__init__()
        self.conv1=nn.Conv2d(in_channels=3,out_channels=1,kernel_size=3,stride=1,padding=0,bias=False)
        nn.init.constant_(self.conv1.weight,1)
        nn.init.constant_(self.conv1.weight[0,0,1,1],-8)
        nn.init.constant_(self.conv1.weight[0,1,1,1],-8)
        nn.init.constant_(self.conv1.weight[0,2,1,1],-8)
      
    def forward(self,x1):
        edge_map=self.conv1(x1)
        return edge_map