(self, n_bodyparts)
| 282 | |
| 283 | class SkeletonTracker(BaseTracker): |
| 284 | def __init__(self, n_bodyparts): |
| 285 | super().__init__(dim=n_bodyparts * 2, dim_z=n_bodyparts) |
| 286 | self.kf.Q[self.kf.dim_z :, self.kf.dim_z :] *= 10 |
| 287 | self.kf.R[self.kf.dim_z :, self.kf.dim_z :] *= 0.01 |
| 288 | self.kf.P[self.kf.dim_z :, self.kf.dim_z :] *= 1000 |
| 289 | |
| 290 | def update(self, pose): |
| 291 | flat = pose.reshape((-1, 1)) |