fastreid.modeling.backbones.osnet.ConvLayer Class Reference

Basic layers. More...

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

Public Member Functions
	__init__ (self, in_channels, out_channels, kernel_size, bn_norm, stride=1, padding=0, groups=1, IN=False)
	forward (self, x)

Public Attributes
	conv
	bn
	relu

Detailed Description

Basic layers.

Convolution layer (conv + bn + relu).

Definition at line 37 of file osnet.py.

Constructor & Destructor Documentation

__init__()

fastreid.modeling.backbones.osnet.ConvLayer.__init__	(		self,
			in_channels,
			out_channels,
			kernel_size,
			bn_norm,
			stride = 1,
			padding = 0,
			groups = 1,
			IN = False )

Definition at line 40 of file osnet.py.

    ):
        super(ConvLayer, self).__init__()
        self.conv = nn.Conv2d(
            in_channels,
            out_channels,
            kernel_size,
            stride=stride,
            padding=padding,
            bias=False,
            groups=groups
        )
        if IN:
            self.bn = nn.InstanceNorm2d(out_channels, affine=True)
        else:
            self.bn = get_norm(bn_norm, out_channels)
        self.relu = nn.ReLU(inplace=True)
 

Member Function Documentation

forward()

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

Definition at line 67 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.ConvLayer.bn

Definition at line 62 of file osnet.py.

conv

fastreid.modeling.backbones.osnet.ConvLayer.conv

Definition at line 52 of file osnet.py.

relu

fastreid.modeling.backbones.osnet.ConvLayer.relu

Definition at line 65 of file osnet.py.

---

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

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