|
|
@ -31,10 +31,8 @@ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
|
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
|
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
|
|
SOFTWARE.
|
|
|
|
SOFTWARE.
|
|
|
|
'''
|
|
|
|
'''
|
|
|
|
import torch
|
|
|
|
|
|
|
|
from torch import nn
|
|
|
|
from torch import nn
|
|
|
|
from torch.nn.parameter import Parameter
|
|
|
|
import torch.nn.functional as F
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class EcaModule(nn.Module):
|
|
|
|
class EcaModule(nn.Module):
|
|
|
|
"""Constructs a ECA module.
|
|
|
|
"""Constructs a ECA module.
|
|
|
@ -100,7 +98,7 @@ class CecaModule(nn.Module):
|
|
|
|
#pytorch circular padding mode is bugged as of pytorch 1.4
|
|
|
|
#pytorch circular padding mode is bugged as of pytorch 1.4
|
|
|
|
# see https://github.com/pytorch/pytorch/pull/17240
|
|
|
|
# see https://github.com/pytorch/pytorch/pull/17240
|
|
|
|
#implement manual circular padding
|
|
|
|
#implement manual circular padding
|
|
|
|
self.conv = nn.Conv1d(1, 1, kernel_size=k_size, padding = 0, bias=False)
|
|
|
|
self.conv = nn.Conv1d(1, 1, kernel_size=k_size, padding=0, bias=False)
|
|
|
|
self.padding = (k_size - 1) // 2
|
|
|
|
self.padding = (k_size - 1) // 2
|
|
|
|
self.sigmoid = nn.Sigmoid()
|
|
|
|
self.sigmoid = nn.Sigmoid()
|
|
|
|
|
|
|
|
|
|
|
@ -109,7 +107,7 @@ class CecaModule(nn.Module):
|
|
|
|
y = self.avg_pool(x)
|
|
|
|
y = self.avg_pool(x)
|
|
|
|
|
|
|
|
|
|
|
|
#manually implement circular padding, F.pad does not seemed to be bugged
|
|
|
|
#manually implement circular padding, F.pad does not seemed to be bugged
|
|
|
|
y = F.pad(y.view(x.shape[0],1,-1),(self.padding,self.padding),mode='circular')
|
|
|
|
y = F.pad(y.view(x.shape[0], 1, -1), (self.padding, self.padding), mode='circular')
|
|
|
|
|
|
|
|
|
|
|
|
# Two different branches of ECA module
|
|
|
|
# Two different branches of ECA module
|
|
|
|
y = self.conv(y)
|
|
|
|
y = self.conv(y)
|
|
|
|