每日Attention学习4——Spatial Attention Module

最新推荐文章于 2025-02-27 15:13:57 发布

xiongxyowo

最新推荐文章于 2025-02-27 15:13:57 发布

阅读量1k

点赞数 4

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

本文链接：https://blog.csdn.net/qq_40714949/article/details/138615489

版权

模块出处

[link] [code] [MM 21] Complementary Trilateral Decoder for Fast and Accurate Salient Object Detection

模块名称

Spatial Attention Module (SAM)

模块作用

空间注意力

模块结构

在这里插入图片描述

模块代码

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


def conv3x3(in_planes, out_planes, stride=1, padding=1, dilation=1, bias=False):
    "3x3 convolution with padding"
    return nn.Conv2d(in_planes, out_planes, kernel_size=3, stride=stride, padding=padding, dilation=dilation