Keras 自定义层报 ‘Node‘ object has no attribute ‘output_masks‘ 问题

Keras 自定义层报 'Node' object has no attribute 'output_masks' 问题

当我用 Keras 复现 Faster R-CNN (系列文章在这里) 时需要用到自定义层实现 RoiPoolingLayer, 但是发现定义完了之后使用时报 ‘Node’ object has no attribute ‘output_masks’. 我开始以为是哪里弄错了, 但是把 Keras 官方例子 复制下来也会报相同的错误, 所以断定应该不是代码的问题. has no attribute 这样的问题应该是缺少库没有导入造成的, 我就看了一下 Keras 的原码, 发现已有的层都是从 engine.base_layer 中的 Layer 继承的, 所以我就加上下面这一句

import keras.engine.base_layer # 增加的
from keras.engine.topology import Layer # 官网例子导入的

然后就成了!

但是, 如果其它的库中有 from tensorflow import keras, 且这个库的 import 放到了 import keras.engine.base_layer 之后, 那还是一样会报 ‘Node’ object has no attribute ‘output_masks’

所以要把 import keras.engine.base_layer 放到 from tensorflow import keras 的后面

你可能感兴趣的:(Keras,深度学习,keras)