TransPose
TransPose copied to clipboard
yangsenius
PyTorch Implementation for "TransPose: Keypoint localization via Transformer", ICCV 2021.
hi there. Is it possible to get the confidence score for the predicted key point for pre-trained model and custom trained model what modifications are required to get the probability...
``` def generate_target(self, joints, joints_vis): ''' :param joints: [num_joints, 3] :param joints_vis: [num_joints, 3] :return: target, target_weight(1: visible, 0: invisible) ''' target_weight = np.ones((self.num_joints, 1), dtype=np.float32) target_weight[:, 0] = joints_vis[:,...
When i run test.py in my dataset, AP(m) and AR(m) are -1. 
How many epochs was the Transpose hA4 pre-trained model fine-tuned on the MPII dataset to get to the benchmarks in the paper? I am using: the following parameters similar to...
I am trying to do a static quantization of the transpose a4 model in PyTorch but I run into this error. It's coming from the aten::add.out operation. How can I...
Hy nice repository. I'm curious how you can generate a json file from the coco keypoint dataset?. Is there a conversion code? You linked to this [repository](https://github.com/leoxiaobin/deep-high-resolution-net.pytorch#data-preparation), but there I...
Training on COCO train2017 dataset：python tools/train.py --cfg experiments/coco/transpose_r/TP_R_256x192_d256_h1024_enc4_mh8.yaml Testing on COCO val2017 dataset：python tools/test.py --cfg experiments/coco/transpose_r/TP_R_256x192_d256_h1024_enc4_mh8.yaml TEST.USE_GT_BBOX True How to get the result of COCO test-dev？
Hi, when I run `tools/compute_flops.py`, [this assertion](https://github.com/yangsenius/TransPose/blob/main/tools/jit_handles.py#L158) always fails.
Has this been tested on live cam feed or for realtime apps? Thanks
Metadata
Owner
Metadata
PyTorch Implementation for "TransPose: Keypoint localization via Transformer", ICCV 2021.