batch_normalization (bool): 是否使用批归一化
conv_first (bool): conv-bn-active(True) or bn-active-conv (False)层堆叠次序
Returns:
x (tensor): 输出
“”"
conv = keras.layers.Conv2D(num_filters,
kernel_size=kernel_size,
strides=strides,
padding=‘same’,
kernel_initializer=‘he_normal’,
kernel_regularizer=keras.regularizers.l2(1e-4))
x = inputs
if conv_first:
x = conv(x)
if batch_normalization:
x = keras.layers.BatchNormalization()(x)
if activation is not None:
x = keras.layers.Activation(activation)(x)
else:
if batch_normalization:
x = keras.layers.BatchNormalization()(x)
if activation is not None:
x = keras.layers.Activation(activation)(x)
x = conv(x)
return x
def resnet(input_shape,depth,num_classes=10):
“”"ResNet
Arguments:</