U-net基础代码

import torch import torch.nn as nn class ConvBlock(nn.Module): def __init__(self, in_channels, out_channels): super(ConvBlock, self).__init__() self.c
posted @ 2024-01-16 11:41  辛宣  阅读(9)  评论(0编辑  收藏  举报