File size: 1,585 Bytes
bcc0f94 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 |
# Original UNet Paper Configuration
# Input shape [1, 572, 572]
# Output shape [64, 388, 388]
decoder_config:
block4: # [1024, 28, 28]
in_channels: 1024
kernel_size: 2
out_channels: 512
padding: [0, 0]
stride: 2 # [512, 52, 52]
block3: # [512, 52, 52]
in_channels: 512
kernel_size: 2
out_channels: 256
padding: [0, 0]
stride: 2 # [256, 100, 100]
block2: # [256, 100, 100]
in_channels: 256
kernel_size: 2
out_channels: 128
padding: [0, 0]
stride: 2 # [128, 196, 196]
block1: # [128, 196, 196]
in_channels: 128
kernel_size: 2
out_channels: 64
padding: [0, 0]
stride: 2 # [64, 388, 388]
encoder_config:
block1: # [1, 572, 572]
all_padding: false
in_channels: 1
maxpool: true
n_layers: 2
out_channels: 64 # [64, 568/2, 568/2] = [64, 284, 284]
block2: # [64, 568/2, 568/2] = [64, 284, 284]
all_padding: false
in_channels: 64
maxpool: true
n_layers: 2
out_channels: 128 # [128, 280/2, 280/2] = [128, 140, 140]
block3: # [128, 280/2, 280/2] = [128, 140, 140]
all_padding: false
in_channels: 128
maxpool: true
n_layers: 2
out_channels: 256 # [256, 136/2, 136/2] = [256, 68, 68]
block4: # [256, 136/2, 136/2] = [256, 68, 68]
all_padding: false
in_channels: 256
maxpool: true
n_layers: 2
out_channels: 512 # [512, 64/2, 64/2] = [512, 32, 32]
block5: # [512, 64/2, 64/2] = [512, 32, 32]
all_padding: false
in_channels: 512
maxpool: false
n_layers: 2
out_channels: 1024 # [1024, 28, 28]
nclasses: 2 |