kennymckormick / pyskl

A toolbox for skeleton-based action recognition.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

此处预处理的代码略有疑问

seekFire opened this issue · comments

输入GCN的tensor要求的人数是2个,但大多数的NTU动作是单人的,因此是需要把第一个人的keypoint信息复制到tensor中的第二个人的位置上。但此处的代码: pyskl/datasets/pipelines/pose_related.py:
2024-03-20 16-20-23 的屏幕截图
第450行, 此处已经将单人的keypoint tensor和zero tensor进行了concatenate操作,得到的tensor已经是代表两个人的了。那下一行的判断条件里的keypoint.shape[0] 就不可能等于1,这还怎么执行随后的复制操作?

还是说单人的动作,只需要给第二个人的位置填充zero tensor,不需要复制第一个人的信息?