tflite_support.task.processor.DetectionOptions

Options for object detection processor.

tflite_support.task.processor.DetectionOptions(
    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
)

Attributes
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: _DetectionOptionsProto
) -> 'DetectionOptions'

Creates a DetectionOptions object from the given protobuf object.

to_pb2

View source

to_pb2() -> _DetectionOptionsProto

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.

Class Variables
category_name_allowlist	None
category_name_denylist	None
display_names_locale	None
max_results	None
score_threshold	None