小编这次要给大家分享的是如何实现keras .h5转移动端的.tflite文件,文章内容丰富,感兴趣的小伙伴可以来了解一下,希望大家阅读完这篇文章之后能够有所收获。
以前tensorflow有bug 在winodws下无法转,但现在好像没有问题了,代码如下
将keras 下的mobilenet_v2转成了tflite
from keras.backend import clear_session import numpy as np import tensorflow as tf clear_session() np.set_printoptions(suppress=True) input_graph_name = "../models/weights.best_mobilenet224.h6" output_graph_name = input_graph_name[:-3] + '.tflite' converter = tf.lite.TFLiteConverter.from_keras_model_file(model_file=input_graph_name) converter.post_training_quantize = True #在windows平台这个函数有问题,无法正常使用 tflite_model = converter.convert() open(output_graph_name, "wb").write(tflite_model) print ("generate:",output_graph_name)
网页题目:如何实现keras.h5转移动端的.tflite文件-创新互联
文章分享:https://www.cdcxhl.com/article30/dsjcpo.html
成都网站建设公司_创新互联,为您提供外贸网站建设、App设计、网站策划、品牌网站建设、网站内链、网站维护
声明:本网站发布的内容(图片、视频和文字)以用户投稿、用户转载内容为主,如果涉及侵权请尽快告知,我们将会在第一时间删除。文章观点不代表本网站立场,如需处理请联系客服。电话:028-86922220;邮箱:631063699@qq.com。内容未经允许不得转载,或转载时需注明来源: 创新互联