()
| 52 | |
| 53 | |
| 54 | def test_sort_ellipse(): |
| 55 | tracklets = dict() |
| 56 | mot_tracker = trackingutils.SORTEllipse(1, 1, 0.6) |
| 57 | poses = np.random.rand(2, 10, 3) |
| 58 | trackers = mot_tracker.track(poses[..., :2]) |
| 59 | assert trackers.shape == (2, 7) |
| 60 | trackingutils.fill_tracklets(tracklets, trackers, poses, imname=0) |
| 61 | assert all(id_ in tracklets for id_ in trackers[:, -2]) |
| 62 | assert all(np.array_equal(tracklets[n][0], pose) for n, pose in enumerate(poses)) |
| 63 | |
| 64 | |
| 65 | def test_tracking_ellipse(real_assemblies, real_tracklets): |