mesh_out: [bs, multi_n, 10475, 3] gt_mesh: [bs, 1, 10475, 3]
(mesh_out, gt_mesh)
| 93 | |
| 94 | |
| 95 | def batch_rigid_align(mesh_out, gt_mesh): |
| 96 | ''' |
| 97 | mesh_out: [bs, multi_n, 10475, 3] |
| 98 | gt_mesh: [bs, 1, 10475, 3] |
| 99 | ''' |
| 100 | # 获取刚性变换参数 |
| 101 | c, R, t = batch_rigid_transform_3D(mesh_out, gt_mesh) |
| 102 | |
| 103 | # 对A进行变换 |
| 104 | A_aligned = torch.matmul(c * R, mesh_out.transpose(2, 3)).transpose(2, 3) + t.unsqueeze(2) |
| 105 | |
| 106 | return A_aligned |
| 107 | |
| 108 | |
| 109 | def transform_joint_to_other_db(src_joint, src_name, dst_name): |
no test coverage detected