easy_ViTPose
easy_ViTPose copied to clipboard
JunkyByte
Easy and fast 2d human and animal multi pose estimation using SOTA ViTPose [Y. Xu et al., 2022] Real-time performances and multiple skeletons supported.
hi, I have modifed your export function for fp16 infernece. I run into the following issue. Here is the modfied export code ``` pmodel.load_state_dict(ckpt) pmodel.eval() pmodel.cuda() pmodel.half() C, H, W...
Hello, I'm attempting to perform fine tuning with your implementation (I'm using the commit e8e2ad1 from April 24, as I don't need feet key points). Unfortunately I think the loss...
### Description After updating my local copy of the codebase to the latest version (previously working on a version that was 2 months old), I encountered a `ModuleNotFoundError` during runtime....
Hello, I have two questions: 1. I was playing around with your project and noticed, that when I run the model on a videostream, the first frame takes quite some...
I trained the vitpose on dataset halpe, then I used this model to visualise a video, but the points are error, any points even beyond the range of the yolo...
When i use the wholebody model from the offical vitpose models, it seems not match. More errors like : size mismatch for backbone.blocks.9.attn.qkv.weight: copying a param with shape torch.Size([3072, 1024])...
I downloaded the vitpose-huge ckpt form [here](https://onedrive.live.com/?redeem=aHR0cHM6Ly8xZHJ2Lm1zL3UvcyFBaW1CZ1lWN0pqVGxnU2hMTUkta2ttdk5mRl9oP2U9ZEVoR0hl&cid=E534267B85818129&id=E534267B85818129%21168&parId=E534267B85818129%21161&o=OneUp) and tried using python model_split.py --source models/vitpose-h.pth --prefix coco --target models/vit_train However I got this error > File "/home/easy_ViTPose/model_split.py", line 56, in main...
