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

谁践踏了优雅 2022-02-04 13:39 393阅读 0赞

# 错误原因与提示 #

我在使用keras搭建模型时，需要使用TensorFlow夹杂搭建，但是keras无论使用函数式还是序列式，每一层的输出类型都是layer类型 ，所以当你使用非keras函数时，得到的变量类型不是layer类型，因此会报下面的错误：

File "..\lib\site-packages\keras\engine\network.py", line 1325, in build_map
    node = layer._inbound_nodes[node_index]
    AttributeError: 'NoneType' object has no attribute '_inbound_nodes'

# 解决办法 #

解决方法就是将自定义层的输出变为keras layer类型，实现也很简单，就是调用keras中的Lambda函数。

下面以简单的案例举例说明如何解决：

## 方法一 ##

from tensorflow import squeeze
    from keras.layers import Lambda
    
    def squeeze_dim(input):
        output = squeeze(input, axis = [1,2])
    
    def my_model(inputs):
        x = Conv2d(inputs)
        ...
        
        # 只需要使用Lambda调用预定义好的函数即可
        x = Lambda(squeeze_dim, name='sqe_dim')(x)
        
        model = Model(input = inputs, output = x)
        ...

## 方法二 ##

from tensorflow import squeeze
    from keras.layers import Lambda
    
    def my_model(inputs):
        x = Conv2d(inputs)
        ...
        
        # 直接使用Lambda将函数包裹
        x = Lambda(lambda x: squeeze(x, axis = [1,2]) )
        
        model = Model(input = inputs, output = x)
        ...

# 结束 #