有没有办法以hdf5格式获得efficientnet预训练重量

Question

我在efficientnet中获取预先训练的权重时遇到了一些问题。所以我在谷歌上搜索了efficientnet.hdf5，但找不到它。所以，无论如何，是否可以获得hdf5格式的预训练重量。谢谢你。

Answer 1

from tensorflow.keras.applications.efficientnet import EfficientNetB0, EfficientNetB5
model = EfficientNetB0(include_top=True, weights="imagenet", input_tensor=None, input_shape=None, pooling=None, classes=1000, classifier_activation="softmax")
model.summary()
# Keras H5 format(older method)
model.save("my_model.h5")
# TensorFlow SavedModel format(recommended)
model.save("my_model")