反正有没有以hdf5格式获得有效的网络预训练权重？

编程问答 2022-06-11

问题描述

在efficiencynet中获取预先训练的权重有些问题。所以我用谷歌搜索了efficiencynet.hdf5，但找不到它。因此，无论如何，都可以获得hdf5格式的预训练权重。谢谢你。

解决方法

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