TensorFlow/Keras:如何从Keras的应用模块中获取缺失的模型(ResNet101、ResNeXt等)?

2024-09-26 18:12:02 发布

您现在位置:Python中文网/ 问答频道 /正文

在我的(最新的)Keras安装和TensorFlow 1.10kerasapi安装中,缺少一些有文档记录的Keras applications。 我按照建议导入Keras的应用程序模块,并按如下方式使用:

from keras import applications
resnet = applications.ResNeXt101(include_top=False, weights='imagenet', input_shape=(SCALED_HEIGHT, SCALED_WIDTH, 3), pooling=None)

我也试过了

^{pr2}$

但在这两种情况下,我都会得到相同类型的错误:

AttributeError: module 'keras.applications' has no attribute 'ResNeXt101'

打印help(applications)产生:

Help on package keras.applications in keras:

NAME
    keras.applications

PACKAGE CONTENTS
    densenet
    imagenet_utils
    inception_resnet_v2
    inception_v3
    mobilenet
    mobilenet_v2
    mobilenetv2
    nasnet
    resnet50
    vgg16
    vgg19
    xception

FUNCTIONS
    DenseNet121 = wrapper(*args, **kwargs)
    DenseNet169 = wrapper(*args, **kwargs)
    DenseNet201 = wrapper(*args, **kwargs)
    InceptionResNetV2 = wrapper(*args, **kwargs)
    InceptionV3 = wrapper(*args, **kwargs)
    MobileNet = wrapper(*args, **kwargs)
    MobileNetV2 = wrapper(*args, **kwargs)
    NASNetLarge = wrapper(*args, **kwargs)
    NASNetMobile = wrapper(*args, **kwargs)
    ResNet50 = wrapper(*args, **kwargs)
    VGG16 = wrapper(*args, **kwargs)
    VGG19 = wrapper(*args, **kwargs)
    Xception = wrapper(*args, **kwargs)
    keras_modules_injection(base_fun)

这表明模型最初在我的安装中不存在。为什么不?它们也没有打包在TensorFlow的kerasapi中。在

我尝试从Keras applications GitHub repository复制文件并将它们粘贴到site-packages/keras/applications/中,但这会导致以下stacktrace:

File "myscript.py", line 517, in get_fpn
    resnet = applications.resnext.ResNeXt101(include_top=False, weights='imagenet', input_shape=(SCALED_HEIGHT, SCALED_WIDTH, 3), pooling=None)
  File "site-packages/keras/applications/resnet_common.py", line 575, in ResNeXt101
    **kwargs)
  File "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或TensorFlow的默认安装中工作?为什么文件没有解释这一点?在


Tags: inpackagestensorflowsiteargswrapperkwargskeras
1条回答
网友
1楼 · 发布于 2024-09-26 18:12:02
  • 问题原因:

第348行的backend对象是None。 我猜你试过这样的方法:

>>> from keras_applications import resnext
>>> resnext.ResNeXt101(weights=None)

backend信息是从keras.应用通过keras_modules_injection装饰器将其添加到keras\u应用程序。在

https://github.com/keras-team/keras/blob/c658993cf596fbd39cf800873bc457e69cfb0cdb/keras/applications/resnext.py#L17

  • 解决问题的步骤:

确保keras和keras应用程序版本如下:

^{pr2}$

如果没有,请使用

>>pip install  upgrade keras keras-applications

将此请求https://github.com/keras-team/keras/pull/11203/files中的更改更新为site-packages/keras/applications

from keras import applications
resnext = applications.resnext.ResNeXt101(include_top=False, weights=None, input_shape=(299,299,3))
print(type(resnext))

相关问题 更多 >

    热门问题