lisiyao21 / Bailando

Code for CVPR 2022 paper "Bailando: 3D dance generation via Actor-Critic GPT with Choreographic Memory"
Other
382 stars 59 forks source link

About kinetic feature extracting. #46

Open ZeyuLing opened 1 year ago

ZeyuLing commented 1 year ago

Hello, thanks for your great paper and code! When reading your evaluation code, I noticed maybe a mistake in utils/features/utils.py line 129. Is the correct code: v1 = ( positions[i + j][joint_idx]

def calc_average_acceleration( positions, i, joint_idx, sliding_window, frame_time ): current_window = 0 average_acceleration = np.zeros(len(positions[0][joint_idx])) for j in range(-sliding_window, sliding_window + 1): if i + j - 1 < 0 or i + j + 1 >= len(positions): continue v2 = ( positions[i + j + 1][joint_idx] - positions[i + j][joint_idx] ) / frame_time v1 = ( positions[i + j][joint_idx]