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)
Example #2
0
 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)