Tensoflow目标检测实战训练模型转换至tflite并部署
Posted 海里的鱼2022
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Tensoflow目标检测实战训练模型转换至tflite并部署相关的知识,希望对你有一定的参考价值。
官方参考笔记
..../models/research/object_detection/colab_tutorials/convert_odt_model_to_TFLite.ipynb
1. 下载预训练模型或自己数据训练得到的模型
wget http://download.tensorflow.org/models/object_detection/tf2/20200711/ssd_mobilenet_v2_fpnlite_640x640_coco17_tpu-8.tar.gz
以mobilenetV模型为例,解压后文件包括 :
根目录
checkpoint目录(模型主要文件)
2. 输出pb模型文件,inference graph
python object_detection/export_tflite_graph_tf2.py \\
--trained_checkpoint_dir ssd_mobilenet_v2_fpnlite_640x640_coco17_tpu-8/checkpoint \\
--output_directory ssd_mobilenet_v2_fpnlite_640x640_coco17_tpu-8/tflite \\
--pipeline_config_path ssd_mobilenet_v2_fpnlite_640x640_coco17_tpu-8/pipe
以上是关于Tensoflow目标检测实战训练模型转换至tflite并部署的主要内容,如果未能解决你的问题,请参考以下文章
Tensoflow目标检测实战训练模型转换至tflite并部署
Keras深度学习实战(15)——从零开始实现YOLO目标检测