#Rank issue about pyskl HOT 2 CLOSED

Hello, please check this example on how to use the provided script: https://github.com/kennymckormick/pyskl/blob/main/examples/extract_diving48_skeleton/diving48_example.ipynb

from pyskl.

I ran the above mentioned link: @kennymckormick
Traceback (most recent call last):
File "/home/ubuntu/Documents/aiguard/pyskl/tools/data/custom_2d_skeleton.py", line 172, in
main()
File "/home/ubuntu/Documents/aiguard/pyskl/tools/data/custom_2d_skeleton.py", line 145, in main
pose_results = pose_inference(pose_model, frames, det_results)
File "/home/ubuntu/Documents/aiguard/pyskl/tools/data/custom_2d_skeleton.py", line 68, in pose_inference
pose = inference_top_down_pose_model(model, f, d, format='xyxy')[0]
File "/home/ubuntu/miniconda3/envs/aiguard1/lib/python3.7/site-packages/mmcv/utils/misc.py", line 340, in new_func
output = old_func(*args, **kwargs)
File "/home/ubuntu/Documents/aiguard/mmpose/mmpose/apis/inference.py", line 387, in inference_top_down_pose_model
use_multi_frames=use_multi_frames)
File "/home/ubuntu/Documents/aiguard/mmpose/mmpose/apis/inference.py", line 247, in _inference_single_pose_model
data = test_pipeline(data)
File "/home/ubuntu/Documents/aiguard/mmpose/mmpose/datasets/pipelines/shared_transform.py", line 107, in call
data = t(data)
File "/home/ubuntu/Documents/aiguard/mmpose/mmpose/datasets/pipelines/top_down_transform.py", line 289, in call
c = results['center']
KeyError: 'center'
ERROR:torch.distributed.elastic.multiprocessing.api:failed (exitcode: 1) local_rank: 0 (pid: 7369) of binary: /home/ubuntu/miniconda3/envs/aiguard1/bin/python

Can anyone give me some idea to solve this? Thanks!

from pyskl.