yolov8添加ca注意力机制

登高去梯

已于 2024-04-07 09:55:00 修改

阅读量2.1k

点赞数 10

分类专栏： yolov8 文章标签： YOLO 深度学习 pytorch

于 2023-12-05 18:35:08 首次发布

本文链接：https://blog.csdn.net/weixin_44240141/article/details/134813886

版权

创建文件 coordAtt.py

位置：ultralytics/nn/modules/coordAtt.py

######################  CoordAtt  ####     start   by  AI&CV  ###############################
# https://zhuanlan.zhihu.com/p/655475515
import torch
import torch.nn as nn
import torch.nn.functional as F


class h_sigmoid(nn.Module):
    def __init__(self, inplace=True):
        super(h_sigmoid, self).__init__()
        self.relu = nn.ReLU6(inplace=inplace)

    def forward(self, x):
        return self.relu(x + 3) / 6


class h_swish(nn.Module):
    def __init__(self, inplace=True):
        super(h_swish, self).__init__()
        self.sigmoid = h_sigmoid(inplace=inplace)

    def forward(self, x