YOLO目标检测创新改进与实战案例专栏
专栏目录: YOLO有效改进系列及项目实战目录 包含卷积,主干 注意力,检测头等创新机制 以及 各种目标检测分割项目实战案例
专栏链接: YOLO基础解析+创新改进+实战案例
摘要
在计算机视觉领域,模型效率的重要性日益增加。在本文中,我们系统地研究了用于目标检测的神经网络架构设计选择,并提出了几个关键优化以提高效率。首先,我们提出了一种加权双向特征金字塔网络(BiFPN),它允许轻松快速地进行多尺度特征融合;其次,我们提出了一种复合缩放方法,该方法统一缩放了所有主干网络、特征网络以及框/类别预测网络的分辨率、深度和宽度。基于这些优化和更好的主干网络,我们开发了一种新的目标检测器系列,称为EfficientDet,它在广泛的资源约束条件下始终比先前的技术实现了更好的效率。特别是,我们的EfficientDet-D7在COCO test-dev上以单模型和单尺度实现了55.1 AP的最先
创新点
BiFPN(双向特征金字塔网络)的创新点可以从以下几个方面总结:
高效的双向跨尺度连接:BiFPN的核心创新之一是双向跨尺度连接,允许特征在不同层级之间通过自上而下和自下而上的路径进行更全面的信息传递和融合。这与传统的FPN和PAN不同,后者主要采用自上而下的特征传播方式。
加权特征融合:BiFPN为每条连接边引入了可学习的权重,允许模型根据不同特征的重要性自适应地调整融合方式。这种加权方法优化了多尺度特征的融合效果,提高了特征表示的准确性。
不规则特征网络拓扑:BiFPN不局限于固定的网络结构,而是使用神经网络架构搜索技术来寻找最优的特征网络拓扑结构。这提供了更大的灵活性,适应不同的任务和资源约束。
特征调整和特征选择:在特征融合过程中,BiFPN不仅仅是简单地合并特征,而是通过权重调整和动态特征选择,优化融合结果,确保重要特征得到有效利用。
与EfficientDet的结合:BiFPN是EfficientDet架构的一部分,其中EfficientNet作为骨干网络,BiFPN负责特征融合。BiFPN的设计允许模型在保持准确性的同时实现高效率,特别是在模型放大时,BiFPN的深度和宽度根据复合缩放方法进行调整。
yolov8 引入MPDIoU
# 结合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]))
task与yaml配置
详见:https://blog.csdn.net/shangyanaf/article/details/136021981