Module modelpack 

Structs

ModelPackDetectionConfig

Configuration for ModelPack split detection decoder. The quantization is ignored when decoding float models.

Functions

decode_modelpack_det

Decodes ModelPack detection outputs from quantized tensors.

decode_modelpack_float

Decodes ModelPack detection outputs from float tensors. The boxes are expected to be in XYXY format.

decode_modelpack_split_float

Decodes ModelPack split detection outputs from float tensors. The boxes are expected to be in XYWH format.

decode_modelpack_split_quant

Decodes ModelPack split detection outputs from quantized tensors. The boxes are expected to be in XYWH format.

modelpack_segmentation_to_mask

Converts ModelPack segmentation into a 2D mask. The input segmentation is expected to have shape (H, W, num_classes).