模型不能作为目录存储在谷歌驱动器中

Question

我正在尝试将一个模型转换为tflite。我的代码如下：

model = sm.Unet(BACKBONE, encoder_weights='imagenet')
model.compile('Adam', loss=sm.losses.bce_jaccard_loss, metrics=[sm.metrics.iou_score])
history = model.fit(my_generator, validation_data=validation_datagen, steps_per_epoch=50, 
validation_steps=50, epochs=50)
model.save('drive/MyDrive/small_data_model')
converter = tf.lite.TFLiteConverter.from_saved_model('drive/MyDrive/small_data_model')

我首先将创建的模型保存在我的谷歌驱动器中，然后尝试创建转换器。但是我得到了一个错误：
"SavedModel文件不存在于：drive/MyDrive/small_data_model/｛saved_model.pbtt|saved_model.bb｝">

模型被保存为没有扩展名的单个文件，而不是目录。如何用其他方式将其转换为tflite？

Answer 1

tf.lite.TFLiteConverter.from_saved_model(...)可能只支持本地文件系统。因此，您可以在本地复制并运行它。

如消息所示，它抱怨drive/MyDrive/small_data_model/中不存在saved_model.pbtxt和saved_model.pb。请仔细检查。