返回

专注连锁门店会员营销管理系统

线上+线下+进销存+收银+会员
HiShop > Hi小程序 > 小程序开发 > 官方文档 >

小程序通用 AI 推理接口InferenceSession wx.createInferenceSession

2023-03-16 作者:--

  InferenceSession wx.createInferenceSession(Object object)

  基础库 2.30.0 开始支持,低版本需做兼容处理。

  小程序插件:支持,需要小程序基础库版本不低于 2.30.0

  功能描述

  创建 AI 推理 Session。使用前可参考AI指南文档

  参数

  Object object

  属性类型默认值必填说明

  modelstring是模型文件路径,目前只执行后缀为.onnx格式(支持代码包路径,和本地文件系统路径)

  precesionLevelnumber4否推理精度,有效值为 0 - 4。一般来说,使用的precesionLevel等级越低,推理速度越快,但可能会损失精度。推荐开发者在开发时,在效果满足需求时优先使用更低精度以提高推理速度,节约能耗。

  合法值说明

  0使用fp16 存储浮点,fp16计算,Winograd 算法也采取fp16 计算,开启近似math计算

  1使用fp16 存储浮点,fp16计算,禁用 Winograd 算法,开启近似math计算

  2使用fp16 存储浮点,fp32计算,开启 Winograd,开启近似math计算

  3使用fp32 存储浮点,fp32计算,开启 Winograd,开启近似math计算

  4使用fp32 存储浮点,fp32计算,开启 Winograd,关闭近似math计算

  allowQuantizebooleanfalse否是否生成量化模型推理

  allowNPUbooleanfalse否是否使用NPU推理,仅对IOS有效

  typicalShapeObject否输入典型分辨率

  返回值

  InferenceSession

  示例代码

  // 创建会话,加载模型

  const session = wx.createInferenceSession({

  model: `${wx.env.USER_DATA_PATH}/MNIST.onnx`,

  precisionLevel: 4,

  typicalShape:{input1:[1, 3, 224, 224], input2:[1, 1, 224, 224]}, //除非使用动态轴,一般不用显式指定

  allowNPU: false,

  allowQuantize: false

  })

  // 监听error事件

  session.onError(err => {

  console.error(err)

  })

  // 监听模型加载完成事件

  session.onLoad(() => {

  // 运行推理

  // 其中input1, input2, output0 必须与使用的onnx模型中实际的输入输出名字完全一致,不可随意填写。

  // 模型输入输出信息可以通过Netron 打开onnx模型看到

  session.run({

  input1: {

  type: 'float32',

  data: new Float32Array(3 * 224 * 224).buffer,

  shape: [1, 3, 224, 224] // NCHW 顺序

  },

  // 多个input的添加方法,假设第二个input需要数据类型为uint8

  input2: {

  type: 'uint8',

  data: new Uint8Array(224 * 224).buffer,

  shape: [1, 1, 224, 224]

  },

  }).then(res => {

  console.log(res.output0)

  })

  })

  // 销毁Session

  // session完成创建后可以多次调用run进行推理,直到调用`session.destroy()`释放相关内存。

  // 销毁会话

  session.destroy()

线上+线下+进销存+收银+会员门店零售管理好帮手

  • 极速收银管理

    门店能实现快速收银,提升门店经营效率。

  • 进销存管理

    系统自带进销存管理,更好管理库存销售。

  • 软硬件一体化

    配备收银硬件一站式整体服务。

收 银 会员管理 库存管理 商品管理 交接班 数据报表 营销促单 售后处理 多种收款

获取报价
【本站声明】

1、本网站发布的该篇文章,目的在于分享电商知识及传递、交流相关电商信息,以便您学习或了解电商知识,请您不要用于其他用途;
2、该篇文章中所涉及的商标、标识的商品/服务并非来源于本网站,更非本网站提供,与本网站无关,系他人的商品或服务,本网站对于该类商标、标识不拥有任何权利;
3、本网站不对该篇文章中所涉及的商标、标识的商品/服务作任何明示或暗示的保证或担保;
4、本网站不对文章中所涉及的内容真实性、准确性、可靠性负责,仅系客观性描述,如您需要了解该类商品/服务详细的资讯,请您直接与该类商品/服务的提供者联系。

推荐文章

门店系统 获取报价 立即咨询 免费试用