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
The locale to use for display names specified through
the TFLite Model Metadata.
Overrides the ones provided in the model metadata. Results
below this value are rejected.
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
.Methods
create_from_pb2
@classmethodcreate_from_pb2 ( pb2_obj : _DetectionOptionsProto ) -> 'DetectionOptions'
Creates a DetectionOptions
object from the given protobuf object.
to_pb2
to_pb2
()
->
_DetectionOptionsProto
Generates a protobuf object to pass to the C++ layer.
__eq__
__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.
