about axis-angle output

[amituofo1996]

Hi, it is really a nice work. I want to get anxis-angle format result.Following is my code. with torch.no_grad(): pred_rotmat, pred_betas, pred_camera = model(norm_img.to(device)) pred_output = smpl(betas=pred_betas, body_pose=pred_rotmat[:,1:], global_orient=pred_rotmat[:,0].unsqueeze(1), pose2rot=False,return_full_pose=True) pred_vertices = pred_output.vertices pred_fullpose = pred_output.full_pose homogen_matrot = F.pad(pred_rotmat.view(-1,3,3),[0,1]) pred_pose = tgm.rotation_matrix_to_angle_axis(homogen_matrot).contiguous().view(-1, 72)

i got pred_pose as my result,but it is wrong,I don't know why,will somebody could help me!