バッチ正規化のaxis(軸)について

kerasでpix2pixのモデルを実装しようと試みている初学者です。

あるサイトに掲載されているサンプルコードを参考にしています。

そのなかのBatchNormalizationに関する記述で、
BatchNormalization(axis=bn_axis)とあるのですが、
bn_axis=〇〇に相当する箇所がどこにも見当たりません。

bn_axis = 1 のようにあらかじめ記述してあるのが一般的かと思うのですが、
そのあたりに関して何かアドバイスを頂けますと大変助かります。

よろしくお願いいたします。

発生している問題・エラーメッセージ

エラーメッセージ

該当のソースコード

Python
1def conv_block_unet(x, f, name, bn_axis, bn=True, strides=(2,2)):
2    x = LeakyReLU(0.2)(x)
3    x = Conv2D(f, (3,3), strides=strides, name=name, padding='same')(x)
4    if bn: x = BatchNormalization(axis=bn_axis)(x)
5    return x

試したこと

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

1T2R3M4

2019/06/15 01:45

引数とか。

mayumi8

2019/06/16 16:02

アドバイスありがとうございます。下の方の記述で、axis_num=-1,　conv = conv_block_unet(list_encoder[-1], f, name, axis_num)との記述がありましたので、このバッチ正規化の軸は「axis=-1」を使用しているということで良いのかと思いました。