weixin_41857483的博客

import torch.nn as nnimport torch.nn.functional as Fimport torchclass DoubleConv(nn.Module): """ DoubleConv 模块 (convolution => [BN] => ReLU) * 2 连续两次的卷积操作：U-net网络中，下采样和上采样过程，每一层都会连续进行两次卷积操作 """ def __init__(self, in_chann