BaseComplexModelOutputConverter
![Inheritance diagram of BaseComplexModelOutputConverter](../../../_images/inheritance-211aaf2178cfd5f31be75002d1a1d6d9edc7cbba.png)
BaseComplexModelOutputConverter inheritance diagram
- class savant.base.converter.BaseComplexModelOutputConverter(**kwargs)
Base complex model output converter.
- 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
.
- abstract __call__(*output_layers, model, roi)
Converts raw model output tensors to Savant format.
- Parameters:
- Returns:
a combination of
BaseObjectModelOutputConverter
andBaseAttributeModelOutputConverter
outputs:BBox tensor
(class_id, confidence, xc, yc, width, height, [angle])
offset by roi upper left and scaled by roi width and height,list of attributes values with confidences
(attr_name, value, confidence)
- Return type: