TensorToLabelConverter

TensorToLabelConverter inheritance diagram

class savant.converter.TensorToLabelConverter(apply_softmax=False, **kwargs)

Raw classifier output to label converter.

__call__(*output_layers, model, roi)

Converts attribute (complex) model output layer values to (attr_name, label, confidence) tuples.

tensor_format: TensorFormat = 0

Set to CuPy to get the output_layers tensors in the converter call on the GPU as a cupy.ndarray. Or set to NumPy to get tensors on the host as a numpy.ndarray.