mirror
/
MODNet
mirror of https://github.com/ZHKKKe/MODNet.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

undo more whitespace changes

pull/217/head
Richard Brown 2024-05-06 16:28:24 +02:00
parent f95e1236bc
commit 089a830eea
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/modnet/models/modnet.py
View File

@ -56,7 +56,7 @@ class Conv2dIBNormRelu(nn.Module):
class SEBlock(nn.Module): class SEBlock(nn.Module):
""" SE Block Proposed in https://arxiv.org/pdf/1709.01507.pdf """ SE Block Proposed in https://arxiv.org/pdf/1709.01507.pdf
""" """
def __init__(self, in_channels, out_channels, reduction=1): def __init__(self, in_channels, out_channels, reduction=1):
@ -68,7 +68,7 @@ class SEBlock(nn.Module):
nn.Linear(int(in_channels // reduction), out_channels, bias=False), nn.Linear(int(in_channels // reduction), out_channels, bias=False),
nn.Sigmoid() nn.Sigmoid()
) )
def forward(self, x): def forward(self, x):
b, c, _, _ = x.size() b, c, _, _ = x.size()
w = self.pool(x).view(b, c) w = self.pool(x).view(b, c)