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 相关文章推荐
Event altKey,ctrlKey,shiftKey属性解析
Dec 18 Javascript
html的DOM中document对象anchors集合用法实例
Jan 21 Javascript
jquery中表单 多选框的一种巧妙写法
Sep 06 Javascript
在页面中输出当前客户端时间javascript实例代码
Mar 02 Javascript
浅谈javascript中执行环境(作用域)与作用域链
Dec 08 Javascript
基于BootstrapValidator的Form表单验证(24)
Dec 12 Javascript
用vue和node写的简易购物车实现
Apr 25 Javascript
AngularJS使用ng-class动态增减class样式的方法示例
May 18 Javascript
集成vue到jquery/bootstrap项目的方法
Feb 10 jQuery
vue弹窗组件的实现示例代码
Sep 10 Javascript
jquery实现直播弹幕效果
Nov 28 jQuery
js判断非127开头的IP地址的实例代码
Jan 05 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简单图像创建入门实例
2015/06/10 PHP
PHP7创建销毁session的实例方法
2020/02/03 PHP
破除网页鼠标右键被禁用的绝招大全
2006/12/27 Javascript
权威JavaScript 中的内存泄露模式
2007/08/13 Javascript
javascript 事件绑定问题
2011/01/01 Javascript
Ajax请求在数据量大的时候出现超时的解决方法
2014/02/27 Javascript
3kb jQuery代码搞定各种树形选择的实现方法
2016/06/10 Javascript
Javascript中prototype的使用详解
2016/06/18 Javascript
Javascript中toFixed计算错误(依赖银行家舍入法的缺陷)解决方法
2017/08/22 Javascript
js精确的加减乘除实例
2017/11/14 Javascript
jQuery实现定时隐藏对话框的方法分析
2018/02/12 jQuery
js中Object.defineProperty()方法的不详解
2018/07/09 Javascript
实现一个 Vue 吸顶锚点组件方法
2019/07/10 Javascript
Vue2.X和Vue3.0数据响应原理变化的区别
2019/11/07 Javascript
Python异常学习笔记
2015/02/03 Python
在Python中使用base64模块处理字符编码的教程
2015/04/28 Python
关于Python中浮点数精度处理的技巧总结
2017/08/10 Python
python处理csv中的空值方法
2018/06/22 Python
Python----数据预处理代码实例
2019/03/20 Python
selenium 多窗口切换的实现(windows)
2020/01/18 Python
python词云库wordcloud的使用方法与实例详解
2020/02/17 Python
python tkinter实现下载进度条及抖音视频去水印原理
2021/02/07 Python
html5 canvas-1.canvas介绍(hello canvas)
2013/01/07 HTML / CSS
主键(Primary Key)约束和唯一性(UNIQUE)约束的区别
2013/05/29 面试题
室内设计自我鉴定
2013/10/15 职场文书
个人简历中的自我评价怎么写
2014/01/26 职场文书
导游实习生自荐书
2014/01/28 职场文书
希特勒经典演讲稿
2014/05/19 职场文书
离婚协议书怎么写(范本参考)
2014/09/30 职场文书
单位工作证明格式模板
2014/10/04 职场文书
公安机关党的群众路线教育实践活动剖析材料
2014/10/10 职场文书
感恩主题班会教案
2015/08/12 职场文书
资深HR教你写好简历中的自我评价
2019/05/07 职场文书
python保存大型 .mat 数据文件报错超出 IO 限制的操作
2021/05/10 Python
MySQL中几种插入和批量语句实例详解
2021/09/14 MySQL
nginx 添加http_stub_status_module模块
2022/05/25 Servers