summaryrefslogtreecommitdiff
path: root/models/hpm.py
AgeCommit message (Collapse)Author
2021-04-10Merge branch 'master' into python3.8python3.8Jordan Gong
2021-04-10Calculate pose similarity loss and canonical consistency loss of each part ↵Jordan Gong
after pooling
2021-03-12Merge branch 'master' into python3.8Jordan Gong
# Conflicts: # models/hpm.py # models/rgb_part_net.py # utils/configuration.py # utils/triplet_loss.py
2021-03-12Code refactoringJordan Gong
1. Separate FCs and triplet losses for HPM and PartNet 2. Remove FC-equivalent 1x1 conv layers in HPM 3. Support adjustable learning rate schedulers
2021-02-08Merge branch 'master' into python3.8Jordan Gong
# Conflicts: # models/hpm.py # models/layers.py # models/model.py # models/rgb_part_net.py # utils/configuration.py
2021-02-08Code refactoring, modifications and new featuresJordan Gong
1. Decode features outside of auto-encoder 2. Turn off HPM 1x1 conv by default 3. Change canonical feature map size from `feature_channels * 8 x 4 x 2` to `feature_channels * 2 x 16 x 8` 4. Use mean of canonical embeddings instead of mean of static features 5. Calculate static and dynamic loss separately 6. Calculate mean of parts in triplet loss instead of sum of parts 7. Add switch to log disentangled images 8. Change default configuration
2021-01-07Type hint for python version lower than 3.9Jordan Gong
2021-01-02Correct feature dims after disentanglement and HPM backbone removalJordan Gong
1. Features used in HPM is decoded canonical embedding without transpose convolution 2. Decode pose embedding to image for Part Net 3. Backbone seems to be redundant, we can use feature map given by auto-decoder
2020-12-31Implement some parts of RGB-GaitPart wrapperJordan Gong
1. Triplet loss function and weight init function haven't been implement yet 2. Tuplize features returned by auto-encoder for later unpack 3. Correct comment error in auto-encoder 4. Swap batch_size dim and time dim in HPM and PartNet in case of redundant transpose 5. Find backbone problems in HPM and disable it temporarily 6. Make feature structure by HPM consistent to that by PartNet 7. Fix average pooling dimension issue and incorrect view change in HP
2020-12-31Make HPM capable of processing frames in all batchesJordan Gong
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