Structure

Parameter

Conv_1 (Leaky_relu)

in_channels = 1, out_channels = 4, kernel = (3*3), stride = (1, 1), padding = 1

Maxpool_1

kernel = (2*2), stride = (2, 2), padding = 0

Conv_2 (Leaky_relu)

in_channels = 4, out_channels = 8, kernel = (3*3), stride = (1, 1), padding = 1

Maxpool_2

kernel = (2*2), stride = (2, 2), padding = 0

F. interpolate

scale_factor = 2, mode = 'nearest'

Conv_3 (Leaky_relu)

in_channels = 8, out_channels = 4, kernel = (3*3), stride = (1, 1), padding = 1

F.interpolate

scale_factor = 2, mode = 'nearest'

Conv_4 (Leaky_relu)

in_channels = 4, out_channels = 1, kernel = (3*3), stride = (1, 1), padding = 1