final_vis_holdenTraj = vis_traj_holden final_vis_holdenTraj_holden_initRot = vis_traj_holden_initRot final_vis_holdenTraj_holden_initTrans = vis_traj_holden_initTrans """ Select Speak to visualize """ final_vis_speak = [vis_b2speak_speak[i] for i in selectedIdx] """ Select Face to visualize """ #Final slection to draw a subset final_vis_faceData = [vis_faceData[i] for i in selectedIdx] if bVisualize == False: continue #################################################################################### # Visualization """ Visualize Location + Orientation """ glViewer.setPosOnly(final_vis_posData) glViewer.setFaceNormal(final_vis_faceNormalData) glViewer.setBodyNormal(final_vis_bodyNormalData) """Visualize Trajectory""" #glViewer.set_Holden_Trajectory_3(final_vis_holdenTraj, initTrans=final_vis_holdenTraj_holden_initTrans, initRot=final_vis_holdenTraj_holden_initRot) #pred """Visualize Body""" glViewer.set_Holden_Data_73(final_vis_bodyData, ignore_root=False, initRot=final_vis_body_initRot, initTrans=final_vis_body_initTrans, bIsGT=False) """Visualize Speech""" glViewer.setSpeech_binary(final_vis_speak) """Visualize face""" #glViewer.SetFaceParmData(vis_faceData,False) vis_f2face_trans, vis_f2face_rot = utility.ComputeHeadOrientation(
"""Show only turn change time""" frameLeng = posData[0].shape[1] print(turnChange) selectedFrames = [] for f in turnChange: fStart = max(f - 90, 0) fEnd = min(f + 90, frameLeng - 1) selectedFrames += range(fStart, fEnd) for i in range(len(posData)): posData[i] = posData[i][:, selectedFrames] faceNormalData[i] = faceNormalData[i][:, selectedFrames] bodyNormalData[i] = bodyNormalData[i][:, selectedFrames] glViewer.resetFrameLimit() glViewer.setPosOnly(posData) glViewer.setFaceNormal(faceNormalData) glViewer.setBodyNormal(bodyNormalData) # glViewer.init_gl() # """Generate Trajectory in Holden's form by pos and body orientation""" # #traj_list, initTrans_list,initRot_list = ConvertTrajectory_velocityForm(posData,bodyNormalData) # traj_list, initTrans_list,initRot_list = ConvertTrajectory_velocityForm([posData[0]],[bodyNormalData[0]]) # glViewer.set_Holden_Trajectory_3(traj_list, initTrans=initTrans_list, initRot=initRot_list) glViewer.init_gl() # traj_list_seq.append(np.array(traj_list)) acc_avg = float(accuracy_sum) / cnt_sum
""" Select Trajectory to visualize """ #GT: GTTargetTrajectory + inputBuyer, inputSeller # vis_posData = [vis_posData[1], vis_posData[2], vis_posData[3] ] # vis_faceNormalData = [vis_faceNormalData[1], vis_faceNormalData[2], vis_faceNormalData[3] ] # vis_bodyNormalData = [vis_bodyNormalData[1], vis_bodyNormalData[2], vis_bodyNormalData[3] ] #Pred trajectory + inputBuyer, inputSeller vis_posData = [vis_posData[0], vis_posData[2], vis_posData[3] ] vis_faceNormalData = [vis_faceNormalData[0], vis_faceNormalData[2], vis_faceNormalData[3] ] vis_bodyNormalData = [vis_bodyNormalData[0], vis_bodyNormalData[2], vis_bodyNormalData[3] ] if bVisualize==False: continue """ Visualize Location + Orientation """ glViewer.setPosOnly(vis_posData) glViewer.setFaceNormal(vis_faceNormalData) glViewer.setBodyNormal(vis_bodyNormalData) """Visualize Trajectory""" glViewer.set_Holden_Trajectory_3(traj_list_holdenForm, initTrans=pred_initTrans_list, initRot=pred_initRot_list) #pred #glViewer.set_Holden_Trajectory_3(gt_traj_list_holdenForm, initTrans=gt_initTrans_list, initRot=gt_initRot_list) #GT #glViewer.set_Holden_Trajectory_3([ bodyData[0][-7:-4,:], bodyData[1][-7:-4,:], bodyData[2][-7:-4,:] ], initRot=initRot, initTrans= initTrans) """Visualize Body""" #glViewer.set_Holden_Data_73([output_body_np],initTrans=initTrans_list,initRot=initRot_list) glViewer.set_Holden_Data_73(vis_bodyData, ignore_root=False, initRot=vis_bodyGT_initRot, initTrans= vis_bodyGT_initTrans, bIsGT=False) #glViewer.set_Holden_Data_73(vis_bodyData, ignore_root=False, initRot=vis_bodyGT_initRot, initTrans= vis_bodyGT_initTrans, bIsGT=True) """Visualize Speech""" glViewer.setSpeech_binary( vis_speech )