tflite_support.task.processor.ConfidenceMask

View source on GitHub

2D-array representing the confidence mask in row major order.

  tflite_support 
 . 
 task 
 . 
 processor 
 . 
 ConfidenceMask 
 ( 
 value 
 : 
 np 
 . 
 ndarray 
 ) 
 

For each pixel, the value indicates the prediction confidence usually in the [0, 1] range where higher values represent a stronger confidence. Ultimately this is model specific, and other range of values might be used.

Methods

create_from_pb2

View source

  @classmethod 
 
  create_from_pb2 
 ( 
 pb2_obj 
 : 
 _ConfidenceMaskProto 
 , 
 height 
 : 
 int 
 , 
 width 
 : 
 int 
 ) 
 -> 
 'ConfidenceMask' 
 

Creates a ConfidenceMask object from the given protobuf and size.

to_pb2

View source

  to_pb2 
 () 
 -> 
 _ConfidenceMaskProto 
 

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.