YOLOv7改进:BoTNeT(Bottleneck Transformer)

本文介绍了如何将BoTNet应用于YOLOv7,通过将ResNet最后三个bottleneck模块的卷积替换为全局的Multi-Head Self-Attention(MHSA),在COCO上提高了目标检测精度1.2%,并在实例分割任务中对小物体检测有显著提升。文章详细阐述了BoTNet的设计原理和实现步骤,并指出其简洁性使其成为未来视觉任务研究的潜在backbone。
摘要由CSDN通过智能技术生成

 目录

1.简介

2.YOLOv7改进:BoTNeT

2.1增加以下yolov7_botnet.yaml文件

 2.2 common.py配置

2.3 yolo.py配置修改


1.简介

论文地址 

 Paper

 

 

本文提出的BoTNet是一种简单高效的网络,有效的将SA应用到多种视觉任务,如图像识别、目标检测、实例分割任务。通过将ResNet50中最后三个bottleneck模块的空间卷积替换为全局的SA操作,有效提升了基线模型在签署任务上的性能。

Section I
常用的CNN大多采用3x3的卷积核,鉴于卷积操作可以有效的不糊哦局部信息,但是对于一些视觉任务如目标检测、实例分割、关键点检测还需要建立长程依赖关系。比如实例分割中需要收集场景相关的信息才能学习物体之间的关系;那么为了吧局部信息聚合就需要堆叠多个卷积层。但基于non-local操作可以更有效,也不需要堆叠那么多层。
而建模长程依赖对NLP任务也十分重要,Self-Attention可以有效学习每一对实体之间的关联,基于SA组成的Transformer已经成为NLP

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

陈子迩

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值