U-Net:生物医学画像セグメンテーションのための畳み込みネットワークを読んでいて、これをKerasに実装したいと思っています。
U-Netでは、畳み込み層を連結する必要があります。1つは収縮経路にあり、もう1つは拡張経路にあります(論文の図1.1)。
ただし、それらのサイズが一致しないため、連結する前に畳み込み層の出力のサイズを変更する必要があります。
Kerasでこれを行うにはどうすればよいですか?
KerasにはCropping2Dレイヤーがあります:https://keras.io/layers/convolutional/#cropping2d
...
conv_13 = Conv2D(64, (3, 3), padding='same', activation='relu')(conv_12) # has outputsize of 568x568
...
crop_13 = Cropping2D((392, 392))(conv_13) # crop 568x568 to 392x392 symmetrically
merge_91 = Concatenate()([crop_13, upsampled_81) # concatenate both layers with same 2D size
...
最初のサイズ(568x568)を最後にアップサンプリングされたサイズ(392x392)に連結する例。
この記事はインターネットから収集されたものであり、転載の際にはソースを示してください。
侵害の場合は、連絡してください[email protected]
コメントを追加