fastreid.modeling.backbones.osnet.LightConv3x3 Class Reference

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

Public Member Functions
	__init__ (self, in_channels, out_channels, bn_norm)
	forward (self, x)

Public Attributes
	conv1
	conv2
	bn
	relu

Detailed Description

Lightweight 3x3 convolution.
1x1 (linear) + dw 3x3 (nonlinear).

Definition at line 138 of file osnet.py.

Constructor & Destructor Documentation

__init__()

fastreid.modeling.backbones.osnet.LightConv3x3.__init__	(		self,
			in_channels,
			out_channels,
			bn_norm )

Definition at line 143 of file osnet.py.

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

Member Function Documentation

forward()

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

Definition at line 160 of file osnet.py.

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

Member Data Documentation

bn

fastreid.modeling.backbones.osnet.LightConv3x3.bn

Definition at line 157 of file osnet.py.

conv1

fastreid.modeling.backbones.osnet.LightConv3x3.conv1

Definition at line 145 of file osnet.py.

conv2

fastreid.modeling.backbones.osnet.LightConv3x3.conv2

Definition at line 148 of file osnet.py.

relu

fastreid.modeling.backbones.osnet.LightConv3x3.relu

Definition at line 158 of file osnet.py.

---

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

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