进一步优化一个tflite模型
Further optimize a tflite model
到处都有工具可以将keras模型(.pb)作为输入,对其进行优化并将其保存为tflite。 (即,toco 和 tflite 转换)
然而,我感兴趣的是导入现有的 tflite 模型并针对大小对其进行优化。
更具体地说,我希望 this 对象检测模型的大小约为 1 MB(显然要权衡精度)
这是一个可视化模型的漂亮工具:https://lutzroeder.github.io/netron/
我经历过类似的话题,但它似乎已经过时了:
一定有办法做到这一点!
不幸的是,没有官方支持的进一步优化已转换的 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
到处都有工具可以将keras模型(.pb)作为输入,对其进行优化并将其保存为tflite。 (即,toco 和 tflite 转换) 然而,我感兴趣的是导入现有的 tflite 模型并针对大小对其进行优化。
更具体地说,我希望 this 对象检测模型的大小约为 1 MB(显然要权衡精度)
这是一个可视化模型的漂亮工具:https://lutzroeder.github.io/netron/
我经历过类似的话题,但它似乎已经过时了:
一定有办法做到这一点!
不幸的是,没有官方支持的进一步优化已转换的 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