(pairs)
| 73 | |
| 74 | |
| 75 | def check_if_same_size(pairs): |
| 76 | shapes1 = [img1['img'].shape[-2:] for img1, img2 in pairs] |
| 77 | shapes2 = [img2['img'].shape[-2:] for img1, img2 in pairs] |
| 78 | return all(shapes1[0] == s for s in shapes1) and all(shapes2[0] == s for s in shapes2) |
| 79 | |
| 80 | |
| 81 | def get_pred_pts3d(gt, pred, use_pose=False): |