Commit 19187d3d authored by Matej Choma's avatar Matej Choma

wip finishing preparation

parent ad9e70f0
import torch import torch
from torch import nn from torch import nn
from core.base import base_conv2d, base_upsample, ResidualCatConv from core.base import BaseModule, base_conv2d, base_upsample, ResidualCatConv
import math import math
class ChannelAttention(nn.Module): class ChannelAttention(nn.Module):
This source diff could not be displayed because it is too large. You can view the blob instead.
...@@ -23,6 +23,7 @@ ...@@ -23,6 +23,7 @@
"\n", "\n",
"from core.base import BaseModule\n", "from core.base import BaseModule\n",
"from core.modules import UnetBT, BaseUpsampler\n", "from core.modules import UnetBT, BaseUpsampler\n",
"from core.module_unet import UnetCascade\n",
"\n", "\n",
"import numpy as np" "import numpy as np"
] ]
This diff is collapsed.
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment