使Tensorflow库+预先训练的MobileNet尽可能小,以使APK变小

问题描述 投票:1回答:1

要求

我正在开发具有标签功能的照片管理Android应用,它必须能够识别日常的物体/建筑物/动物/植物。当我拍摄显示自行车,房屋和山脉的照片时,我的应用程序会将照片添加到每个类别中。

Tensorflow +预先训练的MobileNet v2完全符合要求,在我的build.gradle中有此依赖关系:

implementation 'org.tensorflow:tensorflow-android:1.2.0'

问题

问题:它使APK增长了十倍,达到32 MB,这对我的目标受众(移动数据计划缓慢/昂贵且没有WiFi / LAN /宽带的人们来说是不可接受的。]

问题:我可以使用哪些技巧使Tensorflow +预先训练的MobileNet v2变得更小?

尤其是如何从Tensorflow中删除标记时未使用的类?在Android上不会进行任何培训,因此,例如,我不需要所有的培训课程。如果需要,我可以重新编译,但是也可以使用其他任何方法。

android tensorflow apk size-reduction
1个回答
0
投票

您可以执行一些选择来减小模式大小,

  1. 将Mobilenet与alpha 0.25一起使用将使模型尺寸减小约4倍较小
  2. 尝试转换为.tflite
  3. 量化的.tflite模型也将缩小4倍

您可以阅读很多here,我假设您的模型在转换后将为4MB。

© www.soinside.com 2019 - 2024. All rights reserved.