summaryrefslogtreecommitdiff
path: root/models/hpm.py
AgeCommit message (Collapse)Author
2020-12-31Bug Fixes in HPM and PartNetJordan Gong
1. Register list of torch.nn.Module to the network using torch.nn.ModuleList 2. Fix operation error in squeeze list of tensor 3. Replace squeeze with view in HP in case batch size is 1
2020-12-30Add pooling options in HPMJordan Gong
According to [1], we can use GAP and GMP together, or one of both in ablation study. [1]Y. Fu et al., “Horizontal pyramid matching for person re-identification,” in Proceedings of the AAAI Conference on Artificial Intelligence, 2019, vol. 33, pp. 8295–8302.
2020-12-27Adopt type hinting generics in standard collections (PEP 585)Jordan Gong
2020-12-24Implement Horizontal Pyramid Matching (HPM)Jordan Gong