kennymckormick / pyskl

A toolbox for skeleton-based action recognition.
Apache License 2.0
911 stars 175 forks source link

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

Open seekFire opened 4 months ago

seekFire commented 4 months ago

输入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,不需要复制第一个人的信息?