如何将 EfficientNet-Lite4 与 tensorflow Lite 结合使用

Question

我尝试在 google colab 上使用 EfficientNet-Lite4 和 tensorflow Lite，但出现错误 name 'efficienetnet_lite4_spec' is not defined 我该怎么办？或者它只适用于 EfficientNet-Lite0？

Answer 1

来自评论

Code change from

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

to

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

has resolved the issue (paraphrased from RJ Adriaansen & zeewa42)