TensorFlow.js 微信小程序插件开始支持模型缓存的方法


Posted in Javascript onFebruary 21, 2020

通常情况下,微信小程序追求的是短小精悍,即开即用,用完即走,适用于一些简单的应用场景。然而,随着微信小程序开放能力的提高,人们发现用微信小程序可以实现越来越多的功能,小程序也越来越复杂,越来越庞大起来。这个可以从小程序的大小限制的变化看出,最开始小程序的大小限制为1M,后来限制为2M,最新微信又给小程序提供了分包加载机制,开发者将小程序划分成不同的子包,用户在使用时按需进行加载,所有分包大小限制提高到8M。

虽然小程序的大小限制已经大大提升,但对于小程序开发者而言,仍然捉襟见肘。随便几个图片资源、js库就可能导致小程序超重,尤其对于人工智能小程序而言,更是如此。现在的深度学习模型,动辄几十M,多则一两百M。这个时候开发人员就需要进行取舍,选择那些模型规模小,但精度不那么高的模型。比如图片分类,我们就不要选择Inception V3或ResNet之类的超大规模模型,而是选择针对移动设备优化的MobileNet,也能取得不错的效果。

不过即使是MobileNet,其模型大小也有好几M,对于精简小程序大小仍是一个很大的负担。一种解决方案是从网络加载模型,不增加小程序的体积,但这并不是一个完美的解决方案,毕竟每次推导都需要从网络下载模型,会有一定的网络延迟。在前端开发中,为了保持系统的流畅,通常会采用一些缓存技巧来避免每次从网络加载图片、JS等文件。那能否将模型也作为资源缓存起来呢?

Google团队显然也意识到了这种需求,先是在TensorFlow.js中增加了对tfjs模型缓存的支持。最近,TensorFlow.js 微信小程序插件也得到了更新,支持微信小程序模型缓存。

模型缓存利用了微信小程序的storage接口,需要注意微信小程序对storage的限制:同一个微信用户,同一个小程序 storage 上限为 10MB。storage 以用户维度隔离,同一台设备上,A 用户无法读取到 B 用户的数据;不同小程序之间也无法互相读写数据。所以我们只能选用小于10M的模型。

启用模型缓存也非常简单,步骤如下:

修改app.json文件,将tfjsPlugin的版本修改为0.0.8.

"plugins": {
 "tfjsPlugin": {
  "version": "0.0.8",
  "provider": "wx6afed118d9e81df9"
 }
}

在app.js中提供localStorageHandler函数.

var fetchWechat = require('fetch-wechat');
var tf = require('@tensorflow/tfjs-core');
var plugin = requirePlugin('tfjsPlugin');
//app.js
App({
 // expose localStorage handler
 globalData: {localStorageIO: plugin.localStorageIO},
 ...
});

在模型加载时加入localStorageHandler逻辑。

const LOCAL_STORAGE_KEY = 'mobilenet_model';
export class MobileNet {
 private model: tfc.GraphModel;
 constructor() { }
 
 
 async load() {
 
 
  const localStorageHandler = getApp().globalData.localStorageIO(LOCAL_STORAGE_KEY);
  try {
   this.model = await tfc.loadGraphModel(localStorageHandler);
  } catch (e) {
   this.model =
    await tfc.loadGraphModel(MODEL_URL);
   this.model.save(localStorageHandler);
  }
 }

和浏览器缓存机制有点不同的是,只有在代码包被清理的时候本地缓存才会被清理。如果需要处理缓存,可以通过 wx.setStorage/wx.setStorageSync、wx.getStorage/wx.getStorageSync、wx.clearStorage/wx.clearStorageSync,wx.removeStorage/wx.removeStorageSync 对本地缓存进行读写和清理。

另外需要注意的是,当前tfjs模型托管在tfhub上,需要翻墙访问。项目中的说明文件也提及了这个问题,给出了解决方案,但那是针对以前托管在谷歌云上的模型,建立了中国国内用户可以访问到的镜像。耐心等待吧,相信Google的开发人员会解决tfhub的镜像问题的。

以上就是TensorFlow.js 微信小程序插件开始支持模型缓存的方法的详细内容,更多关于TensorFlow小程序支持模型缓存请关注三水点靠木其它相关文章!

Javascript 相关文章推荐
jquery last-child 列表最后一项的样式
Jan 22 Javascript
JQuery的一些小应用收集
Mar 27 Javascript
拖动布局之保存布局页面cookies篇
Oct 29 Javascript
改变文件域的样式实现思路同时兼容ie、firefox
Oct 23 Javascript
解析jquery中的ajax缓存问题
Dec 19 Javascript
javascript记录文本框内文字个数检测文字个数变化
Oct 14 Javascript
原生js和jQuery随意改变div属性style的名称和值
Oct 22 Javascript
javascript表单处理具体实现代码(表单、链接、按钮)
May 07 Javascript
Angular中innerHTML标签的样式不起作用的原因解析
Jun 18 Javascript
记一次react前端项目打包优化的方法
Mar 30 Javascript
详解vue 中 scoped 样式作用域的规则
Sep 14 Javascript
vue 中的动态传参和query传参操作
Nov 09 Javascript
微信小程序使用 vant Dialog组件的正确方式
Feb 21 #Javascript
JS实现判断移动端PC端功能
Feb 21 #Javascript
JS获取表格视图所选行号的ids过程解析
Feb 21 #Javascript
微信小程序vant弹窗组件的实现方式
Feb 21 #Javascript
纯js实现无缝滚动功能代码实例
Feb 21 #Javascript
vue中实现动态生成二维码的方法
Feb 21 #Javascript
JS如何把字符串转换成json
Feb 21 #Javascript
You might like
php 上一篇,下一篇文章实现代码与原理说明
2010/05/09 PHP
深入php self与$this的详解
2013/06/08 PHP
解析php中eclipse 用空格替换 tab键
2013/06/24 PHP
推荐:极酷右键菜单
2006/11/29 Javascript
用jquery实现点击栏目背景色改变
2012/12/10 Javascript
浅谈关于JavaScript API设计的一些建议和准则
2015/06/24 Javascript
JavaScript判断页面加载完之后再执行预定函数的技巧
2016/05/17 Javascript
浅析JavaScript中的array数组类型系统
2016/07/18 Javascript
AngularJs expression详解及简单示例
2016/09/01 Javascript
jQuery实现的无限级下拉菜单功能示例
2016/09/12 Javascript
bootstrap折叠调用collapse()后data-parent不生效的快速解决办法
2017/02/23 Javascript
JS实现的点击表头排序功能示例
2017/03/27 Javascript
使用jQuery实现鼠标点击左右按钮滑动切换
2017/08/04 jQuery
Vue中正确使用jQuery的方法
2017/10/30 jQuery
详解解决使用axios发送json后台接收不到的问题
2018/06/27 Javascript
Vue2(三)实现子菜单展开收缩,带动画效果实现方法
2019/04/28 Javascript
如何写好一个vue组件,老夫的一年经验全在这了(推荐)
2019/05/18 Javascript
js 递归json树实现根据子id查父id的方法分析
2019/11/08 Javascript
举例讲解如何在Python编程中进行迭代和遍历
2016/01/19 Python
基于DATAFRAME中元素的读取与修改方法
2018/06/08 Python
Python pandas DataFrame操作的实现代码
2019/06/21 Python
python 同时读取多个文件的例子
2019/07/16 Python
python装饰器代替set get方法实例
2019/12/19 Python
Linux下升级安装python3.8并配置pip及yum的教程
2020/01/02 Python
详解pytorch tensor和ndarray转换相关总结
2020/09/03 Python
利用Opencv实现图片的油画特效实例
2021/02/28 Python
软件测试企业面试试卷
2016/07/13 面试题
测绘工程个人的自我评价
2013/11/23 职场文书
优秀体育委员自荐书
2014/01/31 职场文书
班主任班级寄语大全
2014/04/04 职场文书
预备党员转正考核材料
2014/06/03 职场文书
热情服务标语
2014/10/07 职场文书
2014年干部作风建设总结
2014/10/23 职场文书
小学少先队工作总结2015
2015/05/26 职场文书
交通处罚决定书
2015/06/24 职场文书
python 爬取吉首大学网站成绩单
2021/06/02 Python