手把手教你从.pb转换为.tflite文件 附加不同版本tensorflow对应的TFLite Python API语句

本文主要是介绍手把手教你从.pb转换为.tflite文件 附加不同版本tensorflow对应的TFLite Python API语句,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

引言

很多博客教程都需要安装bazel编译工具,并且编译tensorflow的源码,过程繁琐。本篇博文教你如何用几行代码就实现.pb到.tflite文件转换。

代码

.pb到.tflite文件转换代码先贴上来,随后做详细讲解。

import tensorflow as tfin_path = r'.\yolov3_coco.pb'
out_path = r'.\yolov3_coco.tflite'input_arrays = ["input/input_data"]
input_shapes = {"input/input_data" :[1, 416, 416, 3]}
output_arrays = ["pred_sbbox/concat_2", "pred_mbbox/concat_2", "pred_lbbox/concat_2"]converter = tf.lite.TFLiteConverter.from_frozen_graph(in_path, input_arrays, output_arrays, input_shapes)
tflite_model = converter.convert()
open("myModel.tflite", "wb").write(tflite_model)

这里需要修改的参数有:in_path, out_path, input_arrays,  output_arrays,  input_shapes. 

in_path:输入的.pb文件路径                    out_path:输出的.tflite文件路径

input_arrays:输入的节点名称                 output_arrays:输出的节点名称

input_shapes:输入的节点形状

设置路径倒是难不倒大家,但是节点名称和节点形状的参数哪里找呢?以生成yolov3的.pb文件为例说明。

yolov3的代码中,.pb文件是通过convert_weight.py文件和freeze_graph.py文件。俗话说得好,解铃还须系铃人,参数就可以从中找到。

freeze_graph.py文件可以找到input_arrays以及output_arrays参数

convert_weight.py文件找到了input_shapes

TFLiteConverter语句对应

但你复制代码后,你的tf.contrib.lite.TFLiteConverter这条语句会报错,因为这条语句是针对tensorflow 1.12极其以上版本。

不同tensorflow版本对应的转换语句如下。

在tensorflow 2.0版本中,转化语句变成了命令行, tflite_convert。在终端执行如下命令,即可查看其具体使用方法。

tflite_convert -h

我在Python Terminal当中的执行结果。

参考

将pb转换为tflite文件

Tensorflow Lite 模型转换

 

如果本篇文章有任何可以改进的地方,欢迎大家在下方留言:)

这篇关于手把手教你从.pb转换为.tflite文件 附加不同版本tensorflow对应的TFLite Python API语句的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!


原文地址:
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.chinasem.cn/article/883281

相关文章

Python FastAPI+Celery+RabbitMQ实现分布式图片水印处理系统

《PythonFastAPI+Celery+RabbitMQ实现分布式图片水印处理系统》这篇文章主要为大家详细介绍了PythonFastAPI如何结合Celery以及RabbitMQ实现简单的分布式... 实现思路FastAPI 服务器Celery 任务队列RabbitMQ 作为消息代理定时任务处理完整

Python Websockets库的使用指南

《PythonWebsockets库的使用指南》pythonwebsockets库是一个用于创建WebSocket服务器和客户端的Python库,它提供了一种简单的方式来实现实时通信,支持异步和同步... 目录一、WebSocket 简介二、python 的 websockets 库安装三、完整代码示例1.

揭秘Python Socket网络编程的7种硬核用法

《揭秘PythonSocket网络编程的7种硬核用法》Socket不仅能做聊天室,还能干一大堆硬核操作,这篇文章就带大家看看Python网络编程的7种超实用玩法,感兴趣的小伙伴可以跟随小编一起... 目录1.端口扫描器:探测开放端口2.简易 HTTP 服务器:10 秒搭个网页3.局域网游戏:多人联机对战4.

使用Python实现快速搭建本地HTTP服务器

《使用Python实现快速搭建本地HTTP服务器》:本文主要介绍如何使用Python快速搭建本地HTTP服务器,轻松实现一键HTTP文件共享,同时结合二维码技术,让访问更简单,感兴趣的小伙伴可以了... 目录1. 概述2. 快速搭建 HTTP 文件共享服务2.1 核心思路2.2 代码实现2.3 代码解读3.

Python使用自带的base64库进行base64编码和解码

《Python使用自带的base64库进行base64编码和解码》在Python中,处理数据的编码和解码是数据传输和存储中非常普遍的需求,其中,Base64是一种常用的编码方案,本文我将详细介绍如何使... 目录引言使用python的base64库进行编码和解码编码函数解码函数Base64编码的应用场景注意

Python基于wxPython和FFmpeg开发一个视频标签工具

《Python基于wxPython和FFmpeg开发一个视频标签工具》在当今数字媒体时代,视频内容的管理和标记变得越来越重要,无论是研究人员需要对实验视频进行时间点标记,还是个人用户希望对家庭视频进行... 目录引言1. 应用概述2. 技术栈分析2.1 核心库和模块2.2 wxpython作为GUI选择的优

Linux卸载自带jdk并安装新jdk版本的图文教程

《Linux卸载自带jdk并安装新jdk版本的图文教程》在Linux系统中,有时需要卸载预装的OpenJDK并安装特定版本的JDK,例如JDK1.8,所以本文给大家详细介绍了Linux卸载自带jdk并... 目录Ⅰ、卸载自带jdkⅡ、安装新版jdkⅠ、卸载自带jdk1、输入命令查看旧jdkrpm -qa

Java实现时间与字符串互相转换详解

《Java实现时间与字符串互相转换详解》这篇文章主要为大家详细介绍了Java中实现时间与字符串互相转换的相关方法,文中的示例代码讲解详细,感兴趣的小伙伴可以跟随小编一起学习一下... 目录一、日期格式化为字符串(一)使用预定义格式(二)自定义格式二、字符串解析为日期(一)解析ISO格式字符串(二)解析自定义

Tomcat版本与Java版本的关系及说明

《Tomcat版本与Java版本的关系及说明》:本文主要介绍Tomcat版本与Java版本的关系及说明,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录Tomcat版本与Java版本的关系Tomcat历史版本对应的Java版本Tomcat支持哪些版本的pythonJ

Python如何使用__slots__实现节省内存和性能优化

《Python如何使用__slots__实现节省内存和性能优化》你有想过,一个小小的__slots__能让你的Python类内存消耗直接减半吗,没错,今天咱们要聊的就是这个让人眼前一亮的技巧,感兴趣的... 目录背景:内存吃得满满的类__slots__:你的内存管理小助手举个大概的例子:看看效果如何?1.