每日Attention学习13——Adaptive Feature Fusion

模块出处

[AAAI 23] [link] [code] High-Resolution Iterative Feedback Network for Camoufaged Object Detection

---

模块名称

Adaptive Feature Fusion

---

模块作用

多尺度特征融合

---

模块结构

---

模块代码

import torch
import torch.nn as nn

class AFF(nn.Module):
    def __init__(self, ch_in=32, reduction=16):
        super(AFF, self).__init__()
        self.avg_pool = nn.AdaptiveAvgPool2d(1)
        self.fc = nn.Sequential(
            nn.Linear(ch_in