summaryrefslogtreecommitdiff
path: root/models/part_net.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-30Correct and refine PartNetJordan Gong
1. Let FocalConv block capable of processing frames in all batches 2. Correct input dims of TFA and output dims of HP 3. Change torch.unsqueeze and torch.cat to torch.stack
2020-12-30Combine FPFE and TFA to PartNetJordan Gong
2020-12-30Combine FPFE and TFA to GaitPartJordan Gong