TensorToItemConverter

TensorToItemConverter inheritance diagram

class savant.converter.TensorToItemConverter(**kwargs)

Tensor to item converter.

__call__(*output_layers, model, roi)

Converts output arrays to floats.

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.