首页 > 解决方案 > Keras 中的 ResNet50v2

问题描述

我想在 Keras 中加载预训练的 ResNet50v2 模型。我试过了

keras.applications.resnet_v2.ResNet50V2()

这给出了一个错误

Traceback (most recent call last):
  File "<input>", line 1, in <module>
AttributeError: module 'keras.applications' has no attribute 'resnet_v2'

在搜索该错误时,此答案建议使用keras_applicationspackage. 所以我尝试了

keras_applications.resnet_v2.ResNet50V2()

这给出了错误

Traceback (most recent call last):
  File "<input>", line 1, in <module>
  File "/home/nagabhushan/.local/lib/python3.6/site-packages/keras_applications/resnet_common.py", line 495, in ResNet50V2
    **kwargs)
  File "/home/nagabhushan/.local/lib/python3.6/site-packages/keras_applications/resnet_common.py", line 348, in ResNet
    data_format=backend.image_data_format(),
AttributeError: 'NoneType' object has no attribute 'image_data_format'

在搜索该错误时,此答案建议使用keras.applicationspackage. 我很困惑。如何加载 ResNetv2 模型?

注意:我能够加载 ResNet50。只有 ResNet50v2 有问题

from keras.applications.resnet50 import ResNet50

标签: kerasresnetpre-trained-model

解决方案


import keras

keras.applications.resnet_v2.ResNet50V2()
以上代码在jupyter notebook中执行在
安装Keras之前,请先安装其后端引擎之一TensorFlow

  • 点安装张量流
  • 点安装 keras

在此处输入图像描述


推荐阅读