YOLOV5加入BIFPN层,助力涨点!

首先找到models文件夹中的common.py文件,加入BIFPN层模块

# BiFPN
# 两个特征图add操作
class BiFPN_Add2(nn.Module):
    def __init__(self, c1, c2):
        super(BiFPN_Add2, self).__init__()
        # 设置可学习参数 nn.Parameter的作用是:将一个不可训练的类型Tensor转换成可以训练的类型parameter
        # 并且会向宿主模型注册该参数 成为其一部分 即model.parameters()会包含这个parameter
        # 从而在参数优化的时候可以自动一起优化
        self.w = nn.Parameter(torch.ones(2, dtype=torch.float32), requires_grad=True)
        self.epsilon = 0.0001
        self.conv = nn.Conv2d(c1, c2, kernel_size=1, stride=1, padding=0)
        self.silu = nn.SiLU()

    def forward(self, x):
        w = self.w
        weight = w / (torch.sum(w, dim=0) + self.epsilon)
        return self.conv(self.silu(weight[0] * x[0] + weight[1] * x[1]))


# 三个特征图add操作
class BiFPN_Add3(nn.Module):
    def __init__(self, c1, c2):
        super(BiFPN_Add3, self).__init__()
        self.w = nn.

相关推荐

  1. YOLOV5加入BIFPN助力

    2024-05-14 12:02:02       29 阅读
  2. YOLOv5 更换Neck之 BiFPN

    2024-05-14 12:02:02       23 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-05-14 12:02:02       94 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-05-14 12:02:02       100 阅读
  3. 在Django里面运行非项目文件

    2024-05-14 12:02:02       82 阅读
  4. Python语言-面向对象

    2024-05-14 12:02:02       91 阅读

热门阅读

  1. nginx配置(多个前端)

    2024-05-14 12:02:02       31 阅读
  2. LeetCode例题讲解:781.森林中的兔子

    2024-05-14 12:02:02       25 阅读
  3. 【思维】根号分治

    2024-05-14 12:02:02       32 阅读
  4. Vue3:介绍

    2024-05-14 12:02:02       26 阅读
  5. Android 开机过程画面

    2024-05-14 12:02:02       32 阅读
  6. iOS学习- iOS获取指定月的天数

    2024-05-14 12:02:02       23 阅读
  7. vue3基于elementplus 简单实现表格二次封装

    2024-05-14 12:02:02       30 阅读
  8. 【Vue】vue中动态样式绑定

    2024-05-14 12:02:02       30 阅读
  9. 《创意代码:Processing艺术编程之旅》系列目录

    2024-05-14 12:02:02       32 阅读
  10. ArrayList与LinkedList的区别

    2024-05-14 12:02:02       34 阅读