进一步优化 tflite 模型

Question

到处都有可用的工具将 keras 模型（.pb）作为输入，对其进行优化并将其保存为 tflite。 （即，toco 和 tflite 转换） 然而，我感兴趣的是导入现有的 tflite 模型并优化它的大小。

更具体地说，我希望 this 对象检测模型大小约为 1 MB（显然需要权衡准确性）

这是一个可视化模型的漂亮工具：https://lutzroeder.github.io/netron/

我经历了一个类似的线程，但它似乎已经过时了： Converting .tflite to .pb

一定有办法做到这一点！

Answer 1

遗憾的是，没有官方支持的进一步优化已转换 TFLite 模型的方法。采用 TensorFlow 模型并将其转换为 TFLite 模型，可选地应用优化技术是唯一受支持的路径。

话虽如此，对于您的特定模型，我相信这是源 TensorFlow 模型。 https://storage.googleapis.com/download.tensorflow.org/models/object_detection/ssd_mobilenet_v1_coco_2018_01_28.tar.gz

然后，您可以使用以下指南将其转换为 TFLite。 https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/running_on_mobile_tensorflowlite.md