我可以将所有tensorflow slim模型转换为tflite吗?

dav*_*shm 1 python machine-learning computer-vision tensorflow tensorflow-lite

我正在训练基于张量流的苗条模型,用于自定义数据集上的图像分类.在我投入大量时间训练如此庞大的数据集之前,我想知道是否可以将苗条模型动物园中的所有模型转换为tflite格式.

另外,我知道我可以将自定义超薄模型转换为冻结图形.这是我之后担心的一步,即从我的自定义训练的.pb模型转换为.tflite.

这支持吗?或者是否有人面临尚未解决的转换问题?

谢谢.

小智 5

许多Slim模型可以转换为TFLite,但它不能保证,因为某些模型可能有TFLite不支持的操作.

您可以做的是尝试在训练之前使用Python中的TFLiteConverter将模型转换为TensorFlow Lite .如果转换成功,那么您可以训练您的TF模型并再次转换它.