修正的轻量型UNet网络

import torch.nn as nn
import torch

from tensorboardX import SummaryWriter

'''
2021/6/1
problem
1.nn.Batch
2.Conv2dtranpose-nn.ConvTranspose2d(in_channels=out_c,out_channels=out_c,kernel_size=4,stride=2)
3.dropout(p)
4.deblock (2*in_c,in_c,out_c)??
'''

class block(nn.Module):
    def __init__(self,in_c,out_c):
        super(block,self).__init__()
        self.block = nn.Sequential(
            nn.Conv2d(in_channels=in_c,out_channels=out_c,kernel_size=3,padding=1),
            nn.BatchNorm2d(out_c),
            nn.ReLU(inplace=True),
            nn.Dropout(p=0.3), 
            nn.Conv2d(in_channels=out_c,out_channels=out_c,kernel_size=3,stride=1,padding=1),
            nn.BatchNorm2d(out_c),
            nn.ReLU(inplace=True