简述ECA

ECA"/>

简述ECA

文章目录

• ECA-NET代码学习（pytorch）
• • ECA_module
  • torch.squeeze()和 unsqueeze()
  • • 数据维度扩充unsqueeze()
    • 数据维度压缩squeeze()
    • torch.squeeze()和torch.unsqueeze()
  • ECA-ResNet
  • torch.view()
  • torch.resize()
  • nn.Sequential()

ECA-NET代码学习（pytorch）

文件目录：

• ECANet-master
  • models
    • eca_mobilenetv2.py
    • eca_module.py
    • eca_ns.py
    • eca_resnet.py
• main.py
• paras_flops.py

ECA_module

文中注释部分

import torch
from torch import nn
from torch.nn.parameter import Parameter# ECA模块
class eca_layer(nn.Module):"""Constructs a ECA module.Args:channel: Number of channels of the input feature mapk_size: Adaptive selection of kernel size""