def __init__(self, in_channels, out_channels, group=1):
super(Block, self).__init__()
self.channel = 32
self.b1 = ops.EResidualBlock(self.channel, self.channel, group=1)
self.b2 = ops.EResidualBlock(self.channel, self.channel, group=1)
self.c1 = ops.BasicBlock(self.channel * 2, self.channel, 1, 1, 0)
self.c2 = ops.BasicBlock(self.channel * 3, self.channel, 1, 1, 0)
def __init__(self, in_channels, out_channels, group=1):
super(Block, self).__init__()
self.b1 = ops.EResidualBlock(64, 64, group=group)
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.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)
def __init__(self,
in_channels,
out_channels,
group=1,
act=nn.ReLU(inplace=True)):
super(Block, self).__init__()
self.b1 = ops.EResidualBlock(64, 64, group=group)
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, 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)