tasi.TrajectoryDataset.most_likely_class

TrajectoryDataset.most_likely_class(by='trajectory', broadcast=False)

Get the name of the most probable object class for each pose or trajectory of the dataset

Parameters:

• by (str) – By which object the most likely class should be determined. Possible values: ‘pose’, ‘trajectory’

• broadcast (bool, optional) – Specifies whether the most likely class should be broadcasted to each pose of the dataset. The option only changes the output for trajectories. Defaults to False.

Returns:

Information about the most probable object class.

If by is “pose” and broadcast is “False” or “True” return the most likely object class of each pose. If by is “trajectory” and broadcast is “False” return the most likely object class of each trajectory. If by is “trajectory” and broadcast is “True” return the most likely object class of each trajectory for each pose.

Return type:

pd.Series

Raises:

ValueError – If the value of “by” is neither ‘pose’ nor ‘trajectory’.