Inheritance diagram for fastreid.layers.pooling.FastGlobalAvgPool2d:
Public Member Functions | |
| __init__ (self, flatten=False) | |
| forward (self, x) | |
Public Attributes | |
| flatten | |
Detailed Description
Definition at line 78 of file pooling.py.
Constructor & Destructor Documentation
◆ __init__()
| fastreid.layers.pooling.FastGlobalAvgPool2d.__init__ | ( | self, | |
| flatten = False ) |
Definition at line 79 of file pooling.py.
79 def __init__(self, flatten=False):
80 super(FastGlobalAvgPool2d, self).__init__()
81 self.flatten = flatten
82
Member Function Documentation
◆ forward()
| fastreid.layers.pooling.FastGlobalAvgPool2d.forward | ( | self, | |
| x ) |
Definition at line 83 of file pooling.py.
83 def forward(self, x):
84 if self.flatten:
85 in_size = x.size()
86 return x.view((in_size[0], in_size[1], -1)).mean(dim=2)
87 else:
88 return x.view(x.size(0), x.size(1), -1).mean(-1).view(x.size(0), x.size(1), 1, 1)
89
90
Member Data Documentation
◆ flatten
| fastreid.layers.pooling.FastGlobalAvgPool2d.flatten |
Definition at line 81 of file pooling.py.
The documentation for this class was generated from the following file:
- smreid/fastreid/layers/pooling.py
