Update modules.py

This commit is contained in:
Chao Fan
2022-12-27 13:02:32 +08:00
committed by GitHub
parent 0ada706376
commit 0915e989ca
+5
View File
@@ -151,6 +151,11 @@ class SeparateBNNecks(nn.Module):
class FocalConv2d(nn.Module): class FocalConv2d(nn.Module):
"""
GaitPart: Temporal Part-based Model for Gait Recognition
CVPR2020: https://openaccess.thecvf.com/content_CVPR_2020/papers/Fan_GaitPart_Temporal_Part-Based_Model_for_Gait_Recognition_CVPR_2020_paper.pdf
Github: https://github.com/ChaoFan96/GaitPart
"""
def __init__(self, in_channels, out_channels, kernel_size, halving, **kwargs): def __init__(self, in_channels, out_channels, kernel_size, halving, **kwargs):
super(FocalConv2d, self).__init__() super(FocalConv2d, self).__init__()
self.halving = halving self.halving = halving