Inheritance diagram for fastreid.layers.pooling.AdaptiveAvgMaxPool2d:

Public Member Functions
	__init__ (self)

	forward (self, x)

Public Attributes
	gap

	gmp

Detailed Description

Definition at line 65 of file pooling.py.

Constructor & Destructor Documentation

◆ init()

fastreid.layers.pooling.AdaptiveAvgMaxPool2d.__init__ ( self )

Definition at line 66 of file pooling.py.

    def __init__(self):
        super(AdaptiveAvgMaxPool2d, self).__init__()
        self.gap = FastGlobalAvgPool2d()
        self.gmp = nn.AdaptiveMaxPool2d(1)
 

Member Function Documentation

◆ forward()

fastreid.layers.pooling.AdaptiveAvgMaxPool2d.forward	(		self,
			x )

Definition at line 71 of file pooling.py.

    def forward(self, x):
        avg_feat = self.gap(x)
        max_feat = self.gmp(x)
        feat = avg_feat + max_feat
        return feat
 
 

Member Data Documentation

◆ gap

fastreid.layers.pooling.AdaptiveAvgMaxPool2d.gap

Definition at line 68 of file pooling.py.

◆ gmp

fastreid.layers.pooling.AdaptiveAvgMaxPool2d.gmp

Definition at line 69 of file pooling.py.

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

smreid/fastreid/layers/pooling.py

Public Member Functions

Public Attributes