yolo_x.TensorToBBoxConverter
![Inheritance diagram of TensorToBBoxConverter](../../../_images/inheritance-ff3eb5a7f3f512938c29982d4660be851ed7792a.png)
TensorToBBoxConverter inheritance diagram
- class savant.converter.yolo_x.TensorToBBoxConverter(decode=False, confidence_threshold=0.25, top_k=3000)
YOLOX output to bbox converter.
- __call__(*output_layers, model, roi)
Converts YOLOX detector output layer tensor to bbox tensor.
- Parameters:
- Returns:
BBox tensor (class_id, confidence, xc, yc, width, height, [angle]) offset by roi upper left and scaled by roi width and height
- 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
.