请问一下 data_numpy[0, frame_index, :, m] = pose[0::2]中的:是什么意思呢?
# self.C:通道数, self.T:帧数,self.V:骨骼点数, self.num_person_in:检测到的人数
data_numpy = np.zeros((self.C, self.T, self.V, self.num_person_in))
print(data_numpy)
for frame_info in video_info['data']:
frame_index = frame_info['frame_index']
for m, skeleton_info in enumerate(frame_info["skeleton"]):
if m >= self.num_person_in:
break
pose = skeleton_info['pose']
score = skeleton_info['score']
data_numpy[0, frame_index, :, m] = pose[0::2]
data_numpy[1, frame_index, :, m] = pose[1::2]
data_numpy[2, frame_index, :, m] = score
运行代码之后显示了如下错误,不知道应该如何修改,希望各位能够指点一下迷津
data_numpy[0, frame_index, :, m] = pose[0::2]
ValueError: cannot copy sequence with size 20 to array axis with dimension 18
该回答引用ChatGPT
这段代码中,data_numpy 是一个四维的 NumPy 数组,其形状为 (self.C, self.T, self.V, self.num_person_in),其中 self.C 表示通道数,self.T 表示帧数,self.V 表示骨骼点数,self.num_person_in 表示检测到的人数。
data_numpy[0, frame_index, :, m] = pose[0::2] 这行代码的作用是将第 m 个人在第 frame_index 帧时的骨骼点的横坐标(即 x 坐标)存入 data_numpy 的第一通道(0),第 frame_index 帧,第 m 个人的所有骨骼点(即第二个冒号)中。具体来说,pose[0::2] 表示取 pose 数组中所有偶数下标的元素,也就是所有横坐标,因为数组下标从 0 开始,所以实际上是取 0、2、4、6、8、10、12、14、16、18 这 10 个元素。因此,pose[0::2] 应该是一个长度为 10 的一维数组,将其赋值给 data_numpy[0, frame_index, :, m] 的一部分,但是根据错误提示,这部分的长度只有 18,因此会出现赋值错误的情况。
要解决这个问题,可以检查一下 data_numpy 的形状是否正确,以及 pose 的长度是否符合预期。如果 pose 的长度不足 20,可以考虑在末尾补零或者将 data_numpy 的形状进行调整。