tflite_support.task.processor.ClassificationOptions

Options for classification processor.

tflite_support.task.processor.ClassificationOptions(
    score_threshold: Optional[float] = None,
    category_name_allowlist: Optional[List[str]] = None,
    category_name_denylist: Optional[List[str]] = None,
    display_names_locale: Optional[str] = None,
    max_results: Optional[int] = None
)

display_names_locale The locale to use for display names specified through the TFLite Model Metadata.
max_results The maximum number of top-scored classification results to return.
score_threshold Overrides the ones provided in the model metadata. Results below this value are rejected.
category_name_allowlist If non-empty, classifications whose class name is not in this set will be filtered out. Duplicate or unknown class names are ignored. Mutually exclusive with category_name_denylist.
category_name_denylist If non-empty, classifications whose class name is in this set will be filtered out. Duplicate or unknown class names are ignored. Mutually exclusive with category_name_allowlist.

Methods

create_from_pb2

View source

@classmethod
create_from_pb2(
    pb2_obj: _ClassificationOptionsProto
) -> 'ClassificationOptions'

Creates a ClassificationOptions object from the given protobuf object.

to_pb2

View source

to_pb2() -> _ClassificationOptionsProto

Generates a protobuf object to pass to the C++ layer.

__eq__

View source

__eq__(
    other: Any
) -> bool

Checks if this object is equal to the given object.

Args
other The object to be compared with.

Returns
True if the objects are equal.

category_name_allowlist None
category_name_denylist None
display_names_locale None
max_results None
score_threshold None