Hanson0910 / RetinaTrack

The pytorch implement of the retina track original paper link: https://arxiv.org/abs/2003.13870
Apache License 2.0
45 stars 9 forks source link

您好,请问一下代码里面的SSH类模块作用是什么? #5

Open Kingpopen opened 2 years ago

Kingpopen commented 2 years ago

class SSH(nn.Module):

def __init__(self, in_channel, out_channel):
    super(SSH, self).__init__()
    assert out_channel % 4 == 0
    leaky = 0

    if (out_channel <= 64):
        leaky = 0.1

    self.conv3X3 = conv_bn_no_relu(in_channel, out_channel//2, stride=1)

    self.conv5X5_1 = conv_bn(in_channel, out_channel//4, stride=1, leaky = leaky)
    self.conv5X5_2 = conv_bn_no_relu(out_channel//4, out_channel//4, stride=1)

    self.conv7X7_2 = conv_bn(out_channel//4, out_channel//4, stride=1, leaky = leaky)
    self.conv7x7_3 = conv_bn_no_relu(out_channel//4, out_channel//4, stride=1)
Hanson0910 commented 2 years ago

class SSH(nn.Module):

def __init__(self, in_channel, out_channel):
    super(SSH, self).__init__()
    assert out_channel % 4 == 0
    leaky = 0

    if (out_channel <= 64):
        leaky = 0.1

    self.conv3X3 = conv_bn_no_relu(in_channel, out_channel//2, stride=1)

    self.conv5X5_1 = conv_bn(in_channel, out_channel//4, stride=1, leaky = leaky)
    self.conv5X5_2 = conv_bn_no_relu(out_channel//4, out_channel//4, stride=1)

    self.conv7X7_2 = conv_bn(out_channel//4, out_channel//4, stride=1, leaky = leaky)
    self.conv7x7_3 = conv_bn_no_relu(out_channel//4, out_channel//4, stride=1)

Single Stage Headless,参考retinaface

Kingpopen commented 2 years ago

class SSH(nn.Module):

def __init__(self, in_channel, out_channel):
    super(SSH, self).__init__()
    assert out_channel % 4 == 0
    leaky = 0

    if (out_channel <= 64):
        leaky = 0.1

    self.conv3X3 = conv_bn_no_relu(in_channel, out_channel//2, stride=1)

    self.conv5X5_1 = conv_bn(in_channel, out_channel//4, stride=1, leaky = leaky)
    self.conv5X5_2 = conv_bn_no_relu(out_channel//4, out_channel//4, stride=1)

    self.conv7X7_2 = conv_bn(out_channel//4, out_channel//4, stride=1, leaky = leaky)
    self.conv7x7_3 = conv_bn_no_relu(out_channel//4, out_channel//4, stride=1)

Single Stage Headless,参考retinaface

好的 十分感谢!

Kingpopen commented 2 years ago

class SSH(nn.Module):

def __init__(self, in_channel, out_channel):
    super(SSH, self).__init__()
    assert out_channel % 4 == 0
    leaky = 0

    if (out_channel <= 64):
        leaky = 0.1

    self.conv3X3 = conv_bn_no_relu(in_channel, out_channel//2, stride=1)

    self.conv5X5_1 = conv_bn(in_channel, out_channel//4, stride=1, leaky = leaky)
    self.conv5X5_2 = conv_bn_no_relu(out_channel//4, out_channel//4, stride=1)

    self.conv7X7_2 = conv_bn(out_channel//4, out_channel//4, stride=1, leaky = leaky)
    self.conv7x7_3 = conv_bn_no_relu(out_channel//4, out_channel//4, stride=1)

Single Stage Headless,参考retinaface

好的 十分感谢!