states.LearnedProcess
class LearnedProcess:Predict step that applies a learned module to every tracklet's field.
Parameters
| Name | Type | Description |
|---|---|---|
| field | str | Tracklet field to propagate. |
| module | callable | Callable ``(field_tensor: (N, D), dt: float) -> (N, D)`` — e.g. a ``torch.nn.Module`` implementing a recurrent query update. Invoked once per frame on all live tracklets. |
Members
Source: unitrack/states/learned.py:29