SEBlock test tmp
This commit is contained in:
parent
4f07b74cc5
commit
3e4058821e
|
@ -470,7 +470,7 @@ local function unet_conv(backend, n_input, n_middle, n_output, se)
|
|||
model:add(SpatialConvolution(backend, n_middle, n_output, 3, 3, 1, 1, 0, 0))
|
||||
model:add(nn.LeakyReLU(0.1, true))
|
||||
if se then
|
||||
model:add(SEBlock(backend, n_output, 4))
|
||||
model:add(SEBlock(backend, n_output, 8))
|
||||
model:add(w2nn.ScaleTable())
|
||||
end
|
||||
return model
|
||||
|
@ -486,7 +486,7 @@ function srcnn.upcunet(backend, ch)
|
|||
block2:add(unet_branch(backend, block1, backend, 128, 128, 4))
|
||||
block2:add(unet_conv(backend, 128, 64, 64, true))
|
||||
local model = nn.Sequential()
|
||||
model:add(unet_conv(backend, ch, 32, 64, false))
|
||||
model:add(unet_conv(backend, ch, 32, 64, true))
|
||||
model:add(unet_branch(backend, block2, backend, 64, 64, 16))
|
||||
model:add(SpatialConvolution(backend, 64, 64, 3, 3, 1, 1, 0, 0))
|
||||
model:add(nn.LeakyReLU(0.1))
|
||||
|
@ -536,7 +536,7 @@ function srcnn.cunet(backend, ch)
|
|||
block2:add(unet_conv(backend, 128, 64, 64, true))
|
||||
|
||||
local model = nn.Sequential()
|
||||
model:add(unet_conv(backend, ch, 32, 64, false))
|
||||
model:add(unet_conv(backend, ch, 32, 64, true))
|
||||
model:add(unet_branch(backend, block2, backend, 64, 64, 16))
|
||||
model:add(SpatialConvolution(backend, 64, 64, 3, 3, 1, 1, 0, 0))
|
||||
model:add(nn.LeakyReLU(0.1))
|
||||
|
|
Loading…
Reference in a new issue