def __init__(self, in_channels, out_channels, group=1): super(Block, self).__init__() self.r1 = ops.ResidualBlock(in_channels, out_channels) self.r2 = ops.ResidualBlock(in_channels*2, out_channels*2) self.r3 = ops.ResidualBlock(in_channels*4, out_channels*4) self.g = ops.BasicBlock(in_channels*8, out_channels, 1, 1, 0) self.ca = CALayer(in_channels)
def __init__(self, in_channels, out_channels, group=1): super(Block, self).__init__() self.b1 = ops.ResidualBlock(64, 64) self.b2 = ops.ResidualBlock(64, 64) self.b3 = ops.ResidualBlock(64, 64) self.c1 = ops.BasicBlock(64 * 2, 64, 1, 1, 0) self.c2 = ops.BasicBlock(64 * 3, 64, 1, 1, 0) self.c3 = ops.BasicBlock(64 * 4, 64, 1, 1, 0)
def __init__(self, in_channels, out_channels, group=1): super(Block, self).__init__() self.r1 = ops.ResidualBlock(in_channels, out_channels) self.r2 = ops.ResidualBlock(in_channels * 2, out_channels * 2) self.r3 = ops.ResidualBlock(in_channels * 4, out_channels * 4) self.g = ops.BasicBlock(in_channels * 8, out_channels, 1, 1, 0) # compression unit self.ca = CALayer(in_channels) # laplacian attention
def __init__(self, channel=64, mobile=False, groups=1): super().__init__() if mobile: self.b1 = ops.EResidualBlock(channel, channel, groups=groups) self.b2 = self.b3 = self.b1 else: self.b1 = ops.ResidualBlock(channel, channel) self.b2 = ops.ResidualBlock(channel, channel) self.b3 = ops.ResidualBlock(channel, channel) self.c1 = nn.Conv2d(channel*2, channel, 1, 1, 0) self.c2 = nn.Conv2d(channel*3, channel, 1, 1, 0) self.c3 = nn.Conv2d(channel*4, channel, 1, 1, 0)
def __init__(self, in_channels, out_channels, group=1, act=nn.ReLU(inplace=True)): super(Block, self).__init__() self.b1 = ops.ResidualBlock(64, 64) self.b2 = ops.ResidualBlock(64, 64) self.b3 = ops.ResidualBlock(64, 64) self.c1 = ops.BasicBlock(64*2, 64, 1) self.c2 = ops.BasicBlock(64*3, 64, 1) self.c3 = ops.BasicBlock(64*4, 64, 1) self.act = act
def __init__(self, in_channels, out_channels, group=1): super(Block, self).__init__() self.r1 = ops.Merge_Run_dual(in_channels, out_channels) self.r2 = ops.ResidualBlock(in_channels, out_channels) self.r3 = ops.EResidualBlock(in_channels, out_channels) #self.g = ops.BasicBlock(in_channels, out_channels, 1, 1, 0) self.ca = CALayer(in_channels)