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 实现GridView异步排序、分页的代码
Feb 06 Javascript
使用jQuery的将桌面应用程序引入浏览器
Nov 19 Javascript
浅谈JavaScript Date日期和时间对象
Dec 29 Javascript
js中unicode转码方法详解
Oct 09 Javascript
jquery UI Datepicker时间控件的使用及问题解决
Apr 28 Javascript
Node.js的Web模板引擎ejs的入门使用教程
Jun 06 Javascript
Bootstrap select下拉联动(jQuery cxselect)
Jan 04 Javascript
canvas时钟效果
Feb 16 Javascript
javascript实现日期三级联动下拉框选择菜单
Dec 03 Javascript
js+html5生成自动排列对话框实例
Oct 09 Javascript
Vue中props的使用详解
Jun 15 Javascript
js制作提示框插件
Dec 24 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的历史和优缺点
2006/10/09 PHP
How do I change MySQL timezone?
2008/03/26 PHP
php xml常用函数的集合(比较详细)
2013/06/06 PHP
php导出中文内容excel文件类实例
2015/07/06 PHP
深入浅析php中sprintf与printf函数的用法及区别
2016/01/08 PHP
php读取torrent种子文件内容的方法(测试可用)
2016/05/03 PHP
微信公众号之主动给用户发送消息功能
2019/06/22 PHP
JavaScript Title、alt提示(Tips)实现源码解读
2010/12/12 Javascript
如何使用jQuery来处理图片坏链具体实现步骤
2013/05/02 Javascript
javascript 上下banner替换具体实现
2013/11/14 Javascript
单击某一段文字改写文本颜色
2014/06/06 Javascript
jQuery实现的网页竖向菜单效果代码
2015/08/26 Javascript
CSS中position属性之fixed实现div居中
2015/12/14 Javascript
Bootstrap基本插件学习笔记之按钮(21)
2016/12/08 Javascript
jQuery Validate表单验证插件的基本使用方法及功能拓展
2017/01/04 Javascript
jquery uploadify如何取消已上传成功文件
2017/02/08 Javascript
Angularjs+bootstrap+table多选(全选)支持单击行选中实现编辑、删除功能
2017/03/27 Javascript
vue2.0 子组件改变props值,并向父组件传值的方法
2018/03/01 Javascript
jquery分页插件pagination使用教程
2018/10/23 jQuery
vue实现动态按钮功能
2019/05/13 Javascript
微信小程序用户拒绝授权的处理方法详解
2019/09/20 Javascript
python虚拟环境的安装配置图文教程
2017/10/20 Python
Django 解决由save方法引发的错误
2020/05/21 Python
python 如何停止一个死循环的线程
2020/11/24 Python
Html+Css+Jquery实现左侧滑动拉伸导航菜单栏的示例代码
2020/03/17 HTML / CSS
美国电子产品主要品牌的授权在线零售商:DataVision
2019/03/23 全球购物
Ajax和javascript的区别
2013/07/20 面试题
门卫岗位职责
2013/11/15 职场文书
教育学专业毕业生的自我鉴定
2013/11/26 职场文书
《新型玻璃》教学反思
2014/04/13 职场文书
学生安全责任书
2014/04/15 职场文书
社团活动总结报告
2014/06/27 职场文书
会议欢迎词范文
2015/01/27 职场文书
升职自荐信范文
2015/03/27 职场文书
导游词之永济鹳雀楼
2020/01/16 职场文书
Spring Data JPA框架Repository自定义实现
2022/04/28 Java/Android