tasi.TrajectoryDataset.get_by_object_class#

TrajectoryDataset.get_by_object_class(object_class)[source]#

Return only the poses of a specific object class.

Parameters:

object_class (ObjectClass) – The object class.

Returns:

Dataset containing only the poses of a defined object class.

Return type:

ObjectDataset

Note

The object class of a pose is determined by the mean probability of all poses in the trajectory.