分类：tflite

2020
09-29

tensorflow pb to tflite 精度下降详解

之前希望在手机端使用深度模型做OCR，于是尝试在手机端部署tensorflow模型，用于图像分类。思路主要是想使用tflite部署到安卓端，但是在使用tflite的时候发现模型的精度大幅度下降，已经不能支持业务需求了，最后就把OCR模型调用写在服务端了，但是精度下降的原因目前也没有找到，现在这里记录一下。工作思路：1.训练图像分类模型；2.模型固化成pb；3.由pb转成tflite文件；但是使用python的tfinterpreter调用tflite文件就已经出... 继续阅读 >
2020
09-29

Pytorch转tflite方式

目标是想把在服务器上用pytorch训练好的模型转换为可以在移动端运行的tflite模型。最直接的思路是想把pytorch模型转换为tensorflow的模型，然后转换为tflite。但是这个转换目前没有发现比较靠谱的方法。经过调研发现最新的tflite已经支持直接从keras模型的转换，所以可以采用keras作为中间转换的桥梁，这样就能充分利用keras高层API的便利性。转换的基本思想就是用pytorch中的各层网络的权重取出来后直接赋值给keras网络中的对应la... 继续阅读 >
2020
09-29

keras .h5转移动端的.tflite文件实现方式

以前tensorflow有bug在winodws下无法转，但现在好像没有问题了,代码如下将keras下的mobilenet_v2转成了tflitefromkeras.backendimportclear_sessionimportnumpyasnpimporttensorflowastfclear_session()np.set_printoptions(suppress=True)input_graph_name="../models/weights.best_mobilenet224.h5"output_graph_name=input_graph_name[:-3]+'.tflite'converter=tf.lite.TFLiteConverter.from_keras_model_fi... 继续阅读 >