java blob转file_人工智能-Object Detection API 模型转tflite并验证

最近在用TensorFlow Object Detection API训练目标检测模型，训练好的pb模型大小17.6M，用pb预测效果不错，于是想着如何在手机端运行，查看官网关于tflite的转换，网址：https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/running_on_mobile_tensorflowlite.md。

预测使用的pb模型：

在主目录下的research目录执行如下：

生成预测使用的pb文件

注意里面的相关路径需要换成自己的ckpt的目录。执行完后，会看到一个名字为--output_directory指定的目录。这里为exported_graphs目录，里面包含了一个SavedModel 和frozen graph，其中的.pb文件就是预测使用的。

如上转换pb和预测过程参考我之前的文章即可。

关于tflite，需要使用如下操作：

生成frozen graph

注意第二行pipeline_config_path为你的pipeline.config目录路径，trained_config_path路径为训练好的ckpt模型文件，output_dir为tflite输出目录文件夹名字，执行完后在tflite目录会看到两个文件tflite_graph.pb and tflite_graph.pbtxt.

接下来就是转tflite了，之前尝试了其他转tflite的方式，虽然只有几行代码，但发现一直有问题，因此按照官网要求一步一步转，这次使用TOCO转tflite，首先需要安装TensorFlow源码并编译。。。。。切记，要不然找不到bazel目录。。。。

编译生成量化的tflite

注意在编译后的bazel同一级目录执行，input_file为上一步生成的tflite_graph.pb文件。output_file为本次要生成的tflite文件。其他的为固定写法，如果你的模型输入不是300x300注意改input_shapes.执行完就能看到一个detect.tflite文件，该文件大小17M多，这次量化后的模型。

float类型的模型转换方式：

生成float(未量化)的tflite

如上生成detect.tflite文件。

注意：在对比tflite和pb的一致性时，输入到tflite的图片需要进行(image-127.5)/128操作。否则模型输出结果和pb文件对不上。

这样pb文件和tflite预测的结果就能一样，不过float模型大小和pb文件差不多，而且量化的时候模型大小为0，不明白为什么量化的时候模型为量化成功原因，现在只能使用float类型的tflite

后续再讲其他转pb和tflite的方式以及验证方式。

喜欢的朋友和点击关注哦

java blob转file_人工智能-Object Detection API 模型转tflite并验证相关推荐

object转float_人工智能-Object Detection API 模型转tflite并验证
最近在用TensorFlow Object Detection API训练目标检测模型,训练好的pb模型大小17.6M,用pb预测效果不错,于是想着如何在手机端运行,查看官网关于tflite的转换,网 ...
tensorflow环境下的识别食物_Tensorflow object detection API 搭建属于自己的物体识别模型——环境搭建与测试...
1.开发环境搭建 ①.安装Anaconda 建议选择 Anaconda3-5.0.1 版本,已经集成大多数库,并将其作为默认python版本(3.6.3),配置好环境变量(Anaconda安装则已经配 ...
Tensorflow object detection API训练自己的目标检测模型详细配置教程 (一)
Tensorflow object detection API 简单介绍Tensorflow object detection API: 这个API是基于tensorflow构造的开源框架,易于构建. ...
转 TensorFlow Object Detection API 多GPU 卡平行计算，加速模型训练速度教学
本篇记录如何使用多张GPU 显示卡,加速TensorFlow Object Detection API 模型训练的过程. 虽然TensorFlow Object Detection API 已经有支援 ...
TensorFlow Object Detection API 多GPU 卡平行计算，加速模型训练速度教学
本篇记录如何使用多张GPU 显示卡,加速TensorFlow Object Detection API 模型训练的过程. 虽然TensorFlow Object Detection API 已经有支援 ...
如何用TF Object Detection API训练交通信号灯检测神经网络？
参加 2019 Python开发者日,请扫码咨询 ↑↑↑ 作者简介:申泽邦(Adam Shan),谷歌认证机器学习专家(Google Developer Expert),兰州大学智能驾驶团队技术负责人 ...
ubunu16.04 TensorFlow object detection API 应用配置
TensorFlow object detection API应用–配置主要参考 : https://github.com/tensorflow/models/blob/master/researc ...
tensorflow object detection API训练公开数据集Oxford-IIIT Pets Dataset
为了避免不必要的麻烦,先说一下我的系统版本 Python 3.6 tensorflow 1.10 windows 7 object detection API安装 object detection A ...
Tensorflow Object detection API 在 Windows10 配置
Tensorflow Object detection API 在 Windows10 下的配置不如在 Ubuntu 下配置方便,但还是有方法的,介绍一下我的配置流程. 官方目标检测的demo中调用了 ...

java blob转file_人工智能-Object Detection API 模型转tflite并验证

java blob转file_人工智能-Object Detection API 模型转tflite并验证相关推荐

最新文章

热门文章