本发明提供了一种移动端花卉识别模型的构建方法,包括如下步骤:
S10、创建一个ImageNet数据集训练好的浮点类型的卷积神经网络模型;
S20、添加量化运算,即在原始的浮点计算模型中的权重读取和激活输出后插入模拟量化操作;
S30、用花卉数据集训练该卷积神经网络模型,直到模型收敛;
S40、将浮点模型转化为8‑bit整数运算模型,得到花卉识别模型;
S50、使用Bazel构建工具将该花卉识别模型编译成APK安装包。
本发明将用于移动端花卉识别的浮点运算卷积神经网络转化为高效8‑bit整数运算卷积神经网络,减小模型大小的同时缩短模型预测时间,而精度下降却非常低。
Copyright © 2015 科易网 版权所有 闽ICP备07063032号-5