Inheritance diagram for fastreid.modeling.backbones.resnext.Bottleneck:
Public Member Functions | |
| __init__ (self, inplanes, planes, bn_norm, with_ibn, baseWidth, cardinality, stride=1, downsample=None) | |
| forward (self, x) | |
Public Attributes | |
| conv1 | |
| bn1 | |
| conv2 | |
| bn2 | |
| conv3 | |
| bn3 | |
| relu | |
| downsample | |
Static Public Attributes | |
| int | expansion = 4 |
Detailed Description
RexNeXt bottleneck type C
Definition at line 27 of file resnext.py.
Constructor & Destructor Documentation
◆ __init__()
| fastreid.modeling.backbones.resnext.Bottleneck.__init__ | ( | self, | |
| inplanes, | |||
| planes, | |||
| bn_norm, | |||
| with_ibn, | |||
| baseWidth, | |||
| cardinality, | |||
| stride = 1, | |||
| downsample = None ) |
Constructor
Args:
inplanes: input channel dimensionality
planes: output channel dimensionality
baseWidth: base width.
cardinality: num of convolution groups.
stride: conv stride. Replaces pooling layer.
Definition at line 33 of file resnext.py.
34 downsample=None):
35 """ Constructor
36 Args:
37 inplanes: input channel dimensionality
38 planes: output channel dimensionality
39 baseWidth: base width.
40 cardinality: num of convolution groups.
41 stride: conv stride. Replaces pooling layer.
42 """
43 super(Bottleneck, self).__init__()
44
45 D = int(math.floor(planes * (baseWidth / 64)))
46 C = cardinality
47 self.conv1 = nn.Conv2d(inplanes, D * C, kernel_size=1, stride=1, padding=0, bias=False)
48 if with_ibn:
49 self.bn1 = IBN(D * C, bn_norm)
50 else:
51 self.bn1 = get_norm(bn_norm, D * C)
52 self.conv2 = nn.Conv2d(D * C, D * C, kernel_size=3, stride=stride, padding=1, groups=C, bias=False)
53 self.bn2 = get_norm(bn_norm, D * C)
54 self.conv3 = nn.Conv2d(D * C, planes * 4, kernel_size=1, stride=1, padding=0, bias=False)
55 self.bn3 = get_norm(bn_norm, planes * 4)
56 self.relu = nn.ReLU(inplace=True)
57
58 self.downsample = downsample
59
Member Function Documentation
◆ forward()
| fastreid.modeling.backbones.resnext.Bottleneck.forward | ( | self, | |
| x ) |
Definition at line 60 of file resnext.py.
60 def forward(self, x):
61 residual = x
62
63 out = self.conv1(x)
64 out = self.bn1(out)
65 out = self.relu(out)
66
67 out = self.conv2(out)
68 out = self.bn2(out)
69 out = self.relu(out)
70
71 out = self.conv3(out)
72 out = self.bn3(out)
73
74 if self.downsample is not None:
75 residual = self.downsample(x)
76
77 out += residual
78 out = self.relu(out)
79
80 return out
81
82
Member Data Documentation
◆ bn1
| fastreid.modeling.backbones.resnext.Bottleneck.bn1 |
Definition at line 49 of file resnext.py.
◆ bn2
| fastreid.modeling.backbones.resnext.Bottleneck.bn2 |
Definition at line 53 of file resnext.py.
◆ bn3
| fastreid.modeling.backbones.resnext.Bottleneck.bn3 |
Definition at line 55 of file resnext.py.
◆ conv1
| fastreid.modeling.backbones.resnext.Bottleneck.conv1 |
Definition at line 47 of file resnext.py.
◆ conv2
| fastreid.modeling.backbones.resnext.Bottleneck.conv2 |
Definition at line 52 of file resnext.py.
◆ conv3
| fastreid.modeling.backbones.resnext.Bottleneck.conv3 |
Definition at line 54 of file resnext.py.
◆ downsample
| fastreid.modeling.backbones.resnext.Bottleneck.downsample |
Definition at line 58 of file resnext.py.
◆ expansion
|
static |
Definition at line 31 of file resnext.py.
◆ relu
| fastreid.modeling.backbones.resnext.Bottleneck.relu |
Definition at line 56 of file resnext.py.
The documentation for this class was generated from the following file:
- smreid/fastreid/modeling/backbones/resnext.py
