def __init__(self, in_channels, out_channels):
super().__init__()
self.align = conv(in_channels, out_channels, kernel_size=1, padding=0, bn=False)
self.trunk = nn.Sequential(
conv_dw_no_bn(out_channels, out_channels),
conv_dw_no_bn(out_channels, out_channels),
conv_dw_no_bn(out_channels, out_channels)
)
self.conv = conv(out_channels, out_channels, bn=False)
def __init__(self, in_channels, out_channels):#--------512,128
super().__init__()
self.align = conv(in_channels, out_channels, kernel_size=1, padding=0, bn=False)
self.trunk = nn.Sequential(
conv_dw_no_bn(out_channels, out_channels), #kernel_size=3, padding=1, stride=1, dilation=1
conv_dw_no_bn(out_channels, out_channels), #kernel_size=3, padding=1, stride=1, dilation=1
conv_dw_no_bn(out_channels, out_channels) #kernel_size=3, padding=1, stride=1, dilation=1
)
self.conv = conv(out_channels, out_channels, bn=False)
def __init__(self, in_channels, cfg):
super().__init__()
self.align = conv(in_channels,
int(cfg[0]),
kernel_size=1,
padding=0,
bn=False)
# self.select = Selection(cfg[0])
self.trunk = nn.Sequential(conv_dw_no_bn(int(cfg[0]), int(cfg[2])),
conv_dw_no_bn(int(cfg[2]), int(cfg[4])),
conv_dw_no_bn(int(cfg[4]), int(cfg[6])))
self.conv = conv(int(cfg[6]), int(cfg[7]), bn=False)