Yolo8
Functions
Link copied to clipboard
fun calculateHeads(module: Module<FP32, Float>, inputValue: Tensor<FP32, Float>, executionContext: ExecutionContext): HeadOutputs
Runs the YOLO graph and returns all head outputs (small/medium/large) for downstream post-processing.
Link copied to clipboard
suspend fun infer(module: Module<FP32, Float>, input: YoloInput, executionContext: ExecutionContext, reportProgress: suspend (current: Int, total: Int, message: String?) -> Unit = { _, _, _ -> }): List<Detection>
Full inference: forward pass + decode + NMS. Expects the input tensor to be preprocessed to the configured model size (see YoloPreprocess).