有没有办法获得hdf5格式的efficientnet预训练权重？
有没有办法获得hdf5格式的efficientnet预训练权重？
提问于 2020-09-30 14:12:27
我在efficientnet中获取预先训练好的权重时遇到了一些问题。所以我用谷歌搜索了一下efficientnet.hdf5，但是找不到。那么有没有办法得到hdf5格式的预先训练好的权重呢？谢谢你。
回答 1
Stack Overflow用户
发布于 2021-01-07 18:40:32
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")页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/64131836
复制相关文章
相似问题
腾讯云开发者
