TensorToVectorConverter

TensorToVectorConverter inheritance diagram

class savant.converter.TensorToVectorConverter(**kwargs)

Tensor to vector converter.

Eg ReID model.

__call__(*output_layers, model, roi)

Converts output array to Python list.

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.