Skip to content

createImageClassifier

createImageClassifier(config?, opts?): Promise<ImageClassifier>

Defined in: packages/runntime/src/zoo/tasks/cv/imageClassification.ts:54

Loads the checkpoint, warms the GPU path once, and returns the runner. Needs a device with shader-f16.

ImageClassifierModel = {}

LoadOptions = {}

Promise<ImageClassifier>