From 727e3195e33eb755b8b8bcf1957b16c4723bf46e Mon Sep 17 00:00:00 2001 From: xysong <44894406+xysong1201@users.noreply.github.com> Date: Wed, 9 Jan 2019 15:29:24 +0800 Subject: [PATCH 1/3] Update unet_demo.py --- unet_demo.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/unet_demo.py b/unet_demo.py index 5ae1bb0..db20fab 100644 --- a/unet_demo.py +++ b/unet_demo.py @@ -81,6 +81,11 @@ def __init__(self, in_channels, out_channels, n_class, kernel_size, self.down3 = DownConv(4 * out_channels, 8 * out_channels, kernel_size, padding, stride) + self.down4 = DownConv(8 * out_channels, 16 * out_channels, kernel_size, + padding, stride) + + self.up4 = UpConv(16 * out_channels, 8 * out_channels, 8 * out_channels, + kel_size, padding, stride) self.up3 = UpConv(8 * out_channels, 4 * out_channels, 4 * out_channels, kernel_size, padding, stride) @@ -99,8 +104,10 @@ def forward(self, x): x1 = self.down1(x) x2 = self.down2(x1) x3 = self.down3(x2) + x4 = self.down4(x4) # Decoder - x_up = self.up3(x3, x2) + x_up = self.up4(x4, x3) + x_up = self.up3(x_up, x2) x_up = self.up2(x_up, x1) x_up = self.up1(x_up, x) x_out = F.log_softmax(self.out(x_up), 1) From 33a1b8e900c5dcee5e16ae76cfcd152877f485cd Mon Sep 17 00:00:00 2001 From: xysong <44894406+xysong1201@users.noreply.github.com> Date: Wed, 9 Jan 2019 15:56:39 +0800 Subject: [PATCH 2/3] Update unet_demo.py --- unet_demo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/unet_demo.py b/unet_demo.py index db20fab..9c55e0f 100644 --- a/unet_demo.py +++ b/unet_demo.py @@ -85,7 +85,7 @@ def __init__(self, in_channels, out_channels, n_class, kernel_size, padding, stride) self.up4 = UpConv(16 * out_channels, 8 * out_channels, 8 * out_channels, - kel_size, padding, stride) + kernel_size, padding, stride) self.up3 = UpConv(8 * out_channels, 4 * out_channels, 4 * out_channels, kernel_size, padding, stride) From 0e7ebda02bf3b476d9a986af725f4356a1095d23 Mon Sep 17 00:00:00 2001 From: xysong <44894406+xysong1201@users.noreply.github.com> Date: Wed, 9 Jan 2019 15:59:00 +0800 Subject: [PATCH 3/3] Update unet_demo.py --- unet_demo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/unet_demo.py b/unet_demo.py index 9c55e0f..6d34508 100644 --- a/unet_demo.py +++ b/unet_demo.py @@ -104,7 +104,7 @@ def forward(self, x): x1 = self.down1(x) x2 = self.down2(x1) x3 = self.down3(x2) - x4 = self.down4(x4) + x4 = self.down4(x3) # Decoder x_up = self.up4(x4, x3) x_up = self.up3(x_up, x2)