Unet实现

xxxXA860

已于 2022-04-24 12:56:59 修改

阅读量1.5k

点赞数

分类专栏：总结文章标签： python

于 2022-04-23 15:18:48 首次发布

本文链接：https://blog.csdn.net/weixin_62938319/article/details/124364363

版权

本文介绍了如何在Python环境中实例化并运行Unet网络，通过示例代码展示了具体的操作过程，最终得到了运行结果。

摘要由CSDN通过智能技术生成

实现片段

from  torch import nn
import torch


class u_dowm(nn.Module):
    def __init__(self,in_channel,out_channel):
        super(u_dowm, self).__init__()
        self.convANDbn=nn.Sequential(
            nn.Conv2d(in_channels=in_channel,out_channels=out_channel,kernel_size=3,stride=1,padding=1),
            nn.BatchNorm2d(out_channel),
            nn.ReLU(),
            nn.Conv2d(in_channels=out_channel, out_channels=out_channel, kernel_size=3, stride=1, padding=1),
            nn.BatchNorm2d(out_channel),
            nn.ReLU(),
        )
        self.downsample=nn.Sequential(
            nn.MaxPool2d(2,2)
        )
    def forward(self,x):
        output=self.convANDbn(x)   #output是与上采样的拼接的
        out=self.downsample(output)
        return output,out


class up(nn.Module):
    def