跳到主要内容
版本:4.x

InferenceSession

方法

destroy

销毁 InferenceSession 实例

支持情况:

参考文档

() => void

offError

取消监听模型加载失败事件. 传入指定回调函数则只取消指定回调,不传则取消所有回调

支持情况:

参考文档

(callback?: OnErrorCallback) => void
参数类型
callbackOnErrorCallback

offLoad

取消监听模型加载完成事件

支持情况:

参考文档

(callback?: OnLoadCallback) => void
参数类型
callbackOnLoadCallback

onError

监听模型加载失败事件

支持情况:

参考文档

(callback: OnErrorCallback) => void
参数类型
callbackOnErrorCallback

onLoad

监听模型加载完成事件

支持情况:

参考文档

(callback: OnLoadCallback) => void
参数类型
callbackOnLoadCallback

run

运行推断 需要在 session.onLoad 回调后使用。接口参数为 Tensors 对象,返回 Promise。 一个 InferenceSession 被创建完成后可以重复多次调用 InferenceSession.run(), 直到调用 session.destroy() 进行销毁。

支持情况:

参考文档

(option: Tensors) => Promise<Tensors>
参数类型
optionTensors

参数

Type

Tensor

参数类型说明
shapenumber[]Tensor shape (Tensor 形状,例如 [1, 3, 224, 224] 即表示一个4唯Tensor,每个维度的长度分别为1, 3, 224, 224)
dataArrayBufferTensor 值,一段 ArrayBuffer
typekeyof TypeArrayBuffer 值的类型,合法值有 uint8, int8, uint32, int32, float32

Tensors

参数类型
__index__index

OnErrorCallback

模型加载失败回调函数

(res: TaroGeneral.CallbackResult) => void
参数类型
resTaroGeneral.CallbackResult

OnLoadCallback

模型加载完成回调函数

(res: TaroGeneral.CallbackResult) => void
参数类型
resTaroGeneral.CallbackResult

API 支持度

API微信小程序H5React NativeHarmony
InferenceSession.destroy✔️
InferenceSession.offError✔️
InferenceSession.offLoad✔️
InferenceSession.onError✔️
InferenceSession.onLoad✔️
InferenceSession.run✔️