CoTracker

class grid.model.perception.tracking.cotracker.CoTracker(*args, **kwargs)

CoTracker: Point Tracking Model

This class implements a wrapper for the CoTracker model, which tracks points in video frames in an online manner (without having to look at the entire video).

Credits:

https://github.com/facebookresearch/co-tracker

License:

This code is licensed under the CC-BY-NC 4.0 License.

Parameters:

• queries (Tensor)

• checkpoint (str | None)

• save_results (bool)

__init__(queries, checkpoint=None, save_results=False)

Initialize CoTracker model with user-supplied queries

Parameters:

• queries (Tensor)

• checkpoint (str | None)

• save_results (bool)

Return type:

None

finalize()

Finalize processing of remaining frames

Return type:

Tuple[Tensor | None, Tensor | None]

process_frame(frame)

Process a single video frame

Parameters:

frame (ndarray)

Return type:

Tuple[Tensor | None, Tensor | None]