Represents one Segmentation object in the image segmenter's results.
tflite_support
.
task
.
processor
.
Segmentation
(
height
:
int
,
width
:
int
,
colored_labels
:
List
[
tflite_support
.
task
.
processor
.
ColoredLabel
],
category_mask
:
Optional
[
np
.
ndarray
]
=
None
,
confidence_masks
:
Optional
[
List
[
ConfidenceMask
]]
=
None
)
Attributes
The height of the mask. This is an intrinsic parameter of the model
being used, and does not depend on the input image dimensions.
The width of the mask. This is an intrinsic parameter of the model
being used, and does not depend on the input image dimensions.
Methods
create_from_pb2
@classmethodcreate_from_pb2 ( pb2_obj : _SegmentationProto ) -> 'Segmentation'
Creates a Segmentation
object from the given protobuf object.
to_pb2
to_pb2
()
->
_SegmentationProto
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.
