每日Attention学习6——Context Aggregation Module

最新推荐文章于 2024-10-16 21:49:27 发布

xiongxyowo

最新推荐文章于 2024-10-16 21:49:27 发布

阅读量627

点赞数 7

分类专栏：每日Attention学习文章标签：论文阅读

本文链接：https://blog.youkuaiyun.com/qq_40714949/article/details/138620912

版权

模块出处

[link] [code] [IJCAI 22] Boundary-Guided Camouflaged Object Detection

模块名称

Context Aggregation Module (CAM)

模块作用

增大感受野，全局特征提取

模块结构

在这里插入图片描述

模块代码

import torch
import torch.nn as nn
import torch.nn.functional as F


class ConvBNR(nn.Module):
    def __init__(self, inplanes, planes, kernel_size=3, stride=1, dilation=1, bias=False):
        super(ConvBNR, self).__init__()

        self.block = nn.Sequential(
            nn.Conv2d(inplanes, planes, kernel_size, stride=stride, padding=dilation, dilation=dilation, bias=bias),
            nn.BatchNorm2d(planes),
            nn.ReLU(inplace=True)
        )