search

yohanshin / WHAM

MIT License
718 stars 78 forks source link

question about rich_test_vit.pth #121

Open rabbitchenrui opened 1 month ago

rabbitchenrui commented 1 month ago

thanks for your great work! I use wham test on rich, but I find that in your provide rich_test_vit.pth, these poses' shape are:

test/Gym_010_lunge1/cam_01 torch.Size([301, 72]) test/Gym_010_pushup1/cam_03 torch.Size([451, 72]) test/Gym_011_cooking1/cam_04 torch.Size([1101, 72])

but in rich's provide test bodies, these are 30072, 45072, 1100*72. So how do you get rich_test_vit.pth, can you provide a script about the test data preprocess. thanks!

isarandi commented 1 month ago

For some reason, the first frame is duplicated for those sequences in the pth file. You can check the ['frame_id'] key and see that test/Gym_010_lunge1/cam_01 includes frame_id 4 twice in the beginning, and indeed the annotations for those two also match.

rabbitchenrui commented 2 weeks ago

@isarandi I found that wham provide rich_eval_utils which is not complete... would you provide complete rich_eval_utils.py?