keras name scope - pai-plznw4me/tensorflow_basic GitHub Wiki

keras 에서의 name scope 에 대해 설명해 본다. layer 단에서 name 을 관리한다.

SimpleConvBlock 의 layer name 은 block1 이다. SimpleConvBlock 안에 conv_1 의 이름은 conv_1이다.

tensorflow 와 같이 scope 형태로 적용되지 않는다.

sbc = SimpleConvBlock(16, 3, 1, 'same', 'block1')

class SimpleConvBlock(tf.keras.layers.Layer):
    def __init__(self, filters, kernel_size, strides, padding, name, **kwargs):
            super(SimpleConvBlock, self).__init__(name=name, **kwargs)
            self.conv_1 = Conv2D(filters, kernel_size, strides, padding, activation='relu', name='conv_1',
                                 kernel_initializer='he_normal')
            self.conv_2 = Conv2D(filters, kernel_size, strides, padding, activation='relu', name='conv_2',
                                 kernel_initializer='he_normal')
            self.maxpool = MaxPool2D(strides=2, padding='same', name='maxp1_3')

    def call(self, inputs):
        layer = self.conv_1(inputs)
        layer = self.conv_2(layer)
        layer = self.maxpool(layer)
        return layer