Inheritance diagram for fastreid.modeling.backbones.resnet.BasicBlock:
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 | |
| relu | |
| se | |
| downsample | |
| stride | |
Static Public Attributes | |
| int | expansion = 1 |
Detailed Description
Constructor & Destructor Documentation
◆ __init__()
| fastreid.modeling.backbones.resnet.BasicBlock.__init__ | ( | self, | |
| inplanes, | |||
| planes, | |||
| bn_norm, | |||
| with_ibn = False, | |||
| with_se = False, | |||
| stride = 1, | |||
| downsample = None, | |||
| reduction = 16 ) |
Definition at line 42 of file resnet.py.
43 stride=1, downsample=None, reduction=16):
44 super(BasicBlock, self).__init__()
45 self.conv1 = nn.Conv2d(inplanes, planes, kernel_size=3, stride=stride, padding=1, bias=False)
46 if with_ibn:
47 self.bn1 = IBN(planes, bn_norm)
48 else:
49 self.bn1 = get_norm(bn_norm, planes)
50 self.conv2 = nn.Conv2d(planes, planes, kernel_size=3, stride=1, padding=1, bias=False)
51 self.bn2 = get_norm(bn_norm, planes)
52 self.relu = nn.ReLU(inplace=True)
53 if with_se:
54 self.se = SELayer(planes, reduction)
55 else:
56 self.se = nn.Identity()
57 self.downsample = downsample
58 self.stride = stride
59
Member Function Documentation
◆ forward()
| fastreid.modeling.backbones.resnet.BasicBlock.forward | ( | self, | |
| x ) |
Definition at line 60 of file resnet.py.
60 def forward(self, x):
61 identity = 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
70 if self.downsample is not None:
71 identity = self.downsample(x)
72
73 out += identity
74 out = self.relu(out)
75
76 return out
77
78
Member Data Documentation
◆ bn1
◆ bn2
◆ conv1
◆ conv2
◆ downsample
◆ expansion
|
static |
◆ relu
◆ se
◆ stride
The documentation for this class was generated from the following file:
- smreid/fastreid/modeling/backbones/resnet.py
