(tvec_gt, tvec_pred, batch_size=None, ambiguity=True)
| 31 | |
| 32 | |
| 33 | def translation_angle(tvec_gt, tvec_pred, batch_size=None, ambiguity=True): |
| 34 | # tvec_gt, tvec_pred (B, 3,) |
| 35 | rel_tangle_deg = compare_translation_by_angle(tvec_gt, tvec_pred) |
| 36 | rel_tangle_deg = rel_tangle_deg * 180.0 / np.pi |
| 37 | |
| 38 | if ambiguity: |
| 39 | rel_tangle_deg = torch.min(rel_tangle_deg, (180 - rel_tangle_deg).abs()) |
| 40 | |
| 41 | if batch_size is not None: |
| 42 | rel_tangle_deg = rel_tangle_deg.reshape(batch_size, -1) |
| 43 | |
| 44 | return rel_tangle_deg |
| 45 | |
| 46 | |
| 47 |
no test coverage detected