weixin_41857483的博客

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