From b1fd4dc587ed95d6390c5c4b19ef7976656f2584 Mon Sep 17 00:00:00 2001 From: hahnec Date: Mon, 4 Sep 2023 22:37:08 +0200 Subject: [PATCH] feat(up): generic upscale handling --- models/unet/sloun_model_adapt.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/unet/sloun_model_adapt.py b/models/unet/sloun_model_adapt.py index 7c763e9..1f3e84f 100644 --- a/models/unet/sloun_model_adapt.py +++ b/models/unet/sloun_model_adapt.py @@ -17,7 +17,7 @@ def __init__(self, n_channels, n_classes, bilinear=True): self.down3 = (Down(256, 512)) self.latent_conv = (SlounLatent(512, 512)) factor = 2 if bilinear else 1 - self.up1 = (Up(512//2*3, 256 // factor, bilinear)) + self.up1 = (Up(512//2*3, 256 // factor, bilinear)) if bilinear else (Up(512, 256 // factor, bilinear)) self.up2 = (Up(256, 128 // factor, bilinear)) self.up3 = (Up(128, 64, bilinear)) self.outc = (OutConv(64, n_classes))