Yolo8

class Yolo8(config: YoloConfig = YoloConfig()) : Model<FP32, Float, Tensor<FP32, Float>, Tensor<FP32, Float>> (source)

Constructors

Link copied to clipboard
constructor(config: YoloConfig = YoloConfig())

Functions

Link copied to clipboard
open suspend override fun calculate(module: Module<FP32, Float>, inputValue: Tensor<FP32, Float>, executionContext: ExecutionContext, reportProgress: suspend (current: Int, total: Int, message: String?) -> Unit): Tensor<FP32, Float>
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
open override fun create(executionContext: ExecutionContext): Module<FP32, Float>
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).

Link copied to clipboard
fun model(executionContext: ExecutionContext): Module<FP32, Float>
Link copied to clipboard
open override fun modelCard(): ModelCard