Inheritance diagram for fastreid.modeling.backbones.resnet.Bottleneck:
Public Member Functions | |
| __init__ (self, inplanes, planes, bn_norm, with_ibn=False, with_se=False, stride=1, downsample=None, reduction=16) | |
| forward (self, x) | |
Public Attributes | |
| conv1 | |
| bn1 | |
| conv2 | |
| bn2 | |
| conv3 | |
| bn3 | |
| relu | |
| se | |
| downsample | |
| stride | |
Static Public Attributes | |
| int | expansion = 4 |
Detailed Description
Constructor & Destructor Documentation
◆ __init__()
| fastreid.modeling.backbones.resnet.Bottleneck.__init__ | ( | self, | |
| inplanes, | |||
| planes, | |||
| bn_norm, | |||
| with_ibn = False, | |||
| with_se = False, | |||
| stride = 1, | |||
| downsample = None, | |||
| reduction = 16 ) |
Definition at line 82 of file resnet.py.
83 stride=1, downsample=None, reduction=16):
84 super(Bottleneck, self).__init__()
85 self.conv1 = nn.Conv2d(inplanes, planes, kernel_size=1, bias=False)
86 if with_ibn:
87 self.bn1 = IBN(planes, bn_norm)
88 else:
89 self.bn1 = get_norm(bn_norm, planes)
90 self.conv2 = nn.Conv2d(planes, planes, kernel_size=3, stride=stride,
91 padding=1, bias=False)
92 self.bn2 = get_norm(bn_norm, planes)
93 self.conv3 = nn.Conv2d(planes, planes * self.expansion, kernel_size=1, bias=False)
94 self.bn3 = get_norm(bn_norm, planes * self.expansion)
95 self.relu = nn.ReLU(inplace=True)
96 if with_se:
97 self.se = SELayer(planes * self.expansion, reduction)
98 else:
99 self.se = nn.Identity()
100 self.downsample = downsample
101 self.stride = stride
102
Member Function Documentation
◆ forward()
| fastreid.modeling.backbones.resnet.Bottleneck.forward | ( | self, | |
| x ) |
Definition at line 103 of file resnet.py.
103 def forward(self, x):
104 residual = x
105
106 out = self.conv1(x)
107 out = self.bn1(out)
108 out = self.relu(out)
109
110 out = self.conv2(out)
111 out = self.bn2(out)
112 out = self.relu(out)
113
114 out = self.conv3(out)
115 out = self.bn3(out)
116 out = self.se(out)
117
118 if self.downsample is not None:
119 residual = self.downsample(x)
120
121 out += residual
122 out = self.relu(out)
123
124 return out
125
126
Member Data Documentation
◆ bn1
◆ bn2
◆ bn3
◆ conv1
◆ conv2
◆ conv3
◆ downsample
◆ expansion
|
static |
◆ relu
◆ se
◆ stride
The documentation for this class was generated from the following file:
- smreid/fastreid/modeling/backbones/resnet.py
