Method evaluate

datasets/UBody_MM.py:72–266

(self, outs, cur_sample_idx)

                

    def evaluate(self, outs, cur_sample_idx):
        annots = self.datalist
        sample_num = len(outs)
        eval_result = {
            'pa_mpvpe_all': [],
            'pa_mpvpe_l_hand': [],
            'pa_mpvpe_r_hand': [],
            'pa_mpvpe_hand': [],
            'pa_mpvpe_face': [],
            'mpvpe_all': [],
            'mpvpe_l_hand': [],
            'mpvpe_r_hand': [],
            'mpvpe_hand': [],
            'mpvpe_face': []
        }

        vis = getattr(cfg, 'vis', False)
        vis_save_dir = cfg.vis_dir

        for n in range(sample_num):
            
            out = outs[n]
            mesh_gt = out['smplx_mesh_cam_target']
            mesh_out = out['smplx_mesh_cam']
            cam_trans = out['cam_trans']
            joint_proj = out['smplx_joint_proj']
            img_wh = (out['img_shape'])
            ann_idx = out['gt_ann_idx']
            img_path = []
            for ann_id in ann_idx:
                img_path.append(annots[ann_id]['img_path'])
            # print(img_path)
            eval_result['img_path'] = img_path
            eval_result['ann_idx'] = ann_idx
            
            # MPVPE from all vertices
            joint_gt_body_wo_trans = np.dot(smpl_x.j14_regressor,
                                            mesh_gt).transpose(1,0,2)
            joint_gt_body_proj = project_points_new(
                points_3d=torch.Tensor(joint_gt_body_wo_trans),
                pred_cam=torch.Tensor(cam_trans),
                focal_length=5000,
                camera_center=torch.Tensor(img_wh/2)
            )  # origin image space
            
            
            
            joint_gt_lhand_wo_trans = np.dot(
                smpl_x.orig_hand_regressor['left'], mesh_gt).transpose(1,0,2)
            joint_gt_lhand_proj = project_points_new(
                points_3d=torch.Tensor(joint_gt_lhand_wo_trans),
                pred_cam=torch.Tensor(cam_trans),
                focal_length=5000,
                camera_center=torch.Tensor(img_wh/2)
            )  # origin image space
            joint_gt_rhand_wo_trans = np.dot(
                smpl_x.orig_hand_regressor['left'], mesh_gt).transpose(1,0,2)
            joint_gt_rhand_proj = project_points_new(