From 3e4058821e7197799c6630aeb07cb876e7de16aa Mon Sep 17 00:00:00 2001 From: nagadomi Date: Tue, 30 Oct 2018 14:44:28 +0000 Subject: [PATCH] SEBlock test tmp --- lib/srcnn.lua | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/lib/srcnn.lua b/lib/srcnn.lua index 24a9407..4f82585 100644 --- a/lib/srcnn.lua +++ b/lib/srcnn.lua @@ -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))