keras: AttributeError: 'NoneType' object has no attribute 'image_data_format'

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接: https://blog.csdn.net/nima1994/article/details/99176849

https://keras-zh.readthedocs.io/applications/ 给出了keras applications 中的一些预训练模型,部分包导入方式应为:

import keras_applications
mod =keras_applications.resnet.ResNet101(include_top=True,
    weights=None,
    input_shape=(224,224,3),
    )

此时会出现如题错误:

data_format=backend.image_data_format(),
AttributeError: 'NoneType' object has no attribute 'image_data_format'

改正方法,代码添加四行修改为:

mod = keras_applications.resnet.ResNet101(include_top=True,
                                          weights=None,
                                          input_shape=(224, 224, 3),
                                          backend=keras.backend,
                                          layers=keras.layers,
                                          models=keras.models,
                                          utils=keras.utils
                                          )

为什么这莫改,可以参考源码:

./Anaconda3/Lib/site-packages/keras/applications/resnet50.py
.:/Anaconda3/Lib/site-packages/keras/applications/init.py
keras.applications.resnet50在构建实例的时候调用了装饰器keras_modules_injection,给keras_applications相关类增加参数。

猜你喜欢

转载自blog.csdn.net/nima1994/article/details/99176849