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)