tensorflow - 有没有办法获得 hdf5 格式的高效网络预训练权重？

Question

我在效率网络中获得预先训练的权重时遇到了一些问题。所以我搜索了efficientnet.hdf5，但找不到它。那么无论如何要获得hdf5格式的预训练权重。感谢你。

标签： tensorflowkerasconv-neural-networkefficientnet

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")