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)