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
)
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
__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
|