fastreid.modeling.backbones.osnet.Conv1x1 Class Reference

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

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

1x1 convolution + bn + relu.

Definition at line 74 of file osnet.py.

Constructor & Destructor Documentation

__init__()

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

Definition at line 77 of file osnet.py.

    def __init__(self, in_channels, out_channels, bn_norm, stride=1, groups=1):
        super(Conv1x1, self).__init__()
        self.conv = nn.Conv2d(
            in_channels,
            out_channels,
            1,
            stride=stride,
            padding=0,
            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.Conv1x1.forward	(		self,
			x )

Definition at line 91 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.Conv1x1.bn

Definition at line 88 of file osnet.py.

conv

fastreid.modeling.backbones.osnet.Conv1x1.conv

Definition at line 79 of file osnet.py.

relu

fastreid.modeling.backbones.osnet.Conv1x1.relu

Definition at line 89 of file osnet.py.

---

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

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