fastreid.modeling.backbones.osnet.Conv3x3 Class Reference

Inheritance diagram for fastreid.modeling.backbones.osnet.Conv3x3:

Public Member Functions
	__init__ (self, in_channels, out_channels, bn_norm, stride=1, groups=1)
	forward (self, x)

Public Attributes
	conv
	bn
	relu

Detailed Description

3x3 convolution + bn + relu.

Definition at line 114 of file osnet.py.

Constructor & Destructor Documentation

__init__()

fastreid.modeling.backbones.osnet.Conv3x3.__init__	(		self,
			in_channels,
			out_channels,
			bn_norm,
			stride = 1,
			groups = 1 )

Definition at line 117 of file osnet.py.

    def __init__(self, in_channels, out_channels, bn_norm, stride=1, groups=1):
        super(Conv3x3, self).__init__()
        self.conv = nn.Conv2d(
            in_channels,
            out_channels,
            3,
            stride=stride,
            padding=1,
            bias=False,
            groups=groups
        )
        self.bn = get_norm(bn_norm, out_channels)
        self.relu = nn.ReLU(inplace=True)
 

Member Function Documentation

forward()

fastreid.modeling.backbones.osnet.Conv3x3.forward	(		self,
			x )

Definition at line 131 of file osnet.py.

    def forward(self, x):
        x = self.conv(x)
        x = self.bn(x)
        x = self.relu(x)
        return x
 
 

Member Data Documentation

bn

fastreid.modeling.backbones.osnet.Conv3x3.bn

Definition at line 128 of file osnet.py.

conv

fastreid.modeling.backbones.osnet.Conv3x3.conv

Definition at line 119 of file osnet.py.

relu

fastreid.modeling.backbones.osnet.Conv3x3.relu

Definition at line 129 of file osnet.py.

---

The documentation for this class was generated from the following file:

• smreid/fastreid/modeling/backbones/osnet.py