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)