Unet实现
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_chan.
原创
2022-04-23 15:18:48 ·
1537 阅读 ·
0 评论