python - 如何在 tensorflow Lite 中使用 EfficientNet-Lite4

Question

我尝试在带有 tensorflow Lite 的 google colab 上使用 EfficientNet-Lite4，但出现错误name 'efficienetnet_lite4_spec' is not defined我该怎么办？还是仅适用于 EfficientNet-Lite0？

Answer 1

来自评论

代码更改从

model = image_classifier.create(train_data, model_spec=efficienetnet_lite4_spec,validation_data=validation_data,epochs=300)

到

model = image_classifier.create(train_data, model_spec=model_spec.get('efficientnet_lite4'),validation_data=validation_data,epochs=300)

已解决该问题（转述自 RJ Adriaansen 和 zeewa42）