Tensoflow目标检测实战训练模型转换至tflite并部署

Posted 2022-11-04 海里的鱼2022

官方参考笔记

..../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