我正在训练基于 tensorflow slim 的模型以在自定义数据集上进行图像分类。在投入大量时间训练如此庞大的数据集之前,我想知道是否可以转换 slim model zoo 中所有可用的模型。到 tflite 格式。
此外,我知道我可以将我的自定义 slim 模型转换为卡住图。我担心的是此后的步骤,即从我自定义训练的 .pb 模型转换为 .tflite。
这是否受支持?或者是否有人面临尚未解决的转换问题?
谢谢。
请您参考如下方法:
许多 Slim 模型可以转换为 TFLite,但这并不能保证,因为某些模型可能具有 TFLite 不支持的操作。
您可以做的是尝试使用 TFLiteConverter 将您的模型转换为 TensorFlow Lite在训练前使用 Python。如果转换成功,那么您可以训练您的 TF 模型并再次转换它。
