BaseAttributeModelOutputConverter
![Inheritance diagram of BaseAttributeModelOutputConverter](../../../_images/inheritance-1ba88e081ecec759400d34a764f734c5f4e76e46.png)
BaseAttributeModelOutputConverter inheritance diagram
- class savant.base.converter.BaseAttributeModelOutputConverter(**kwargs)
Base attribute model output converter.
- abstract __call__(*output_layers, model, roi)
Converts raw model output tensors to a list of values in several formats:
- classification
string label
int or float value
- ReID
extracted descriptor as a vector of values
and so on, depending on the task.
- Parameters:
- Returns:
list of attributes values with confidences
(attr_name, value, confidence)
- Return type:
- tensor_format: TensorFormat = 0
Set to
CuPy
to get theoutput_layers
tensors in the converter call on the GPU as acupy.ndarray
. Or set toNumPy
to get tensors on the host as anumpy.ndarray
.