Inheritance diagram for fastreid.layers.pooling.ClipGlobalAvgPool2d:
Public Member Functions | |
| __init__ (self) | |
| forward (self, x) | |
Public Attributes | |
| avgpool | |
Detailed Description
Definition at line 91 of file pooling.py.
Constructor & Destructor Documentation
◆ __init__()
| fastreid.layers.pooling.ClipGlobalAvgPool2d.__init__ | ( | self | ) |
Definition at line 92 of file pooling.py.
92 def __init__(self):
93 super().__init__()
94 self.avgpool = FastGlobalAvgPool2d()
95
Member Function Documentation
◆ forward()
| fastreid.layers.pooling.ClipGlobalAvgPool2d.forward | ( | self, | |
| x ) |
Definition at line 96 of file pooling.py.
96 def forward(self, x):
97 x = self.avgpool(x)
98 x = torch.clamp(x, min=0., max=1.)
99 return x
Member Data Documentation
◆ avgpool
| fastreid.layers.pooling.ClipGlobalAvgPool2d.avgpool |
Definition at line 94 of file pooling.py.
The documentation for this class was generated from the following file:
- smreid/fastreid/layers/pooling.py
