mlrl.common.cython.label_space_info module

@author: Michael Rapp (michael.rapp.ml@gmail.com)

class mlrl.common.cython.label_space_info.LabelSpaceInfo

Bases: object

Provides information about the label space that may be used as a basis for making predictions.

class mlrl.common.cython.label_space_info.LabelVectorSet

Bases: LabelSpaceInfo

Stores a set of unique label vectors, as well as their frequency.

visit(visitor: LabelVectorSetVisitor)

Visits the label vectors and frequencies stored by the set.

Parameters:

visitor – The LabelVectorSetVisitor that should be used to access the label vectors and frequencies

class mlrl.common.cython.label_space_info.LabelVectorSetVisitor

Bases: object

Defines the methods that must be implemented by a visitor that accesses the label vectors and frequencies stored by a LabelVectorSet.

abstract visit_label_vector(label_vector: ndarray, frequency: int)

Must be implemented by subclasses in order to visit label vectors and their frequencies.

Parameters:

  • label_vector – An np.ndarray of type uint8, shape (num_relevant_labels) that stores the indices of the relevant labels

  • frequency – The frequency of the label vector

class mlrl.common.cython.label_space_info.NoLabelSpaceInfo

Bases: LabelSpaceInfo

Does not provide any information about the label space.