tflite_support.task.processor.ClassificationOptions

Options for classification processor.

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

Creates a ClassificationOptions object from the given protobuf object.

to_pb2

View source

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

__eq__

View source

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