transformer优化(二)-DETR 学习笔记

DETR是一种创新的目标检测模型,利用Transformer的编码器-解码器结构,避免了传统方法中的锚框生成和非最大抑制。它直接从图像中预测目标边界框和类别,简化了检测流程。

论文地址:https://arxiv.org/pdf/2005.12872.pdf

代码地址:https://github.com/bubbliiiing/detr-pytorch

https://github.com/facebookresearch/detr

1.是什么?

DETR(Detection Transformer)是一种基于Transformer的目标检测模型。它采用了一种全新的思路,将目标检测问题转化为一个集合预测问题。相比传统的目标检测方法,DETR不需要使用锚框或者候选框,而是直接从输入图像中预测出一组目标的边界框和类别。

DETR的整体结构如下:

  1. 编码器(Encoder):使用一系列的Transformer编码器层来提取输入图像的特征表示。
  2. 解码器(Decoder):使用一系列的Transformer解码器层来生成目标的边界框和类别预测。
  3. 多头自注意力机制(Multi-Head Self-Attention):用于捕捉目标之间的关系和上下文信息。
  4. 线性层和位置编码:用于将特征映射到目标的边界框和类别预测。

DETR的实现思路是通过Transformer模型来实现目标检测任务。它将目标检测问题转化为一个集合预测问题,通过直接从输入图像中预测出一组目标的边界框和类别来完成目标检测任务。

2.为什么?

提出了一种将目标检测视为直接集预测问题的新方法。我们的方法简化了检测管道,有效地消除了许多手工设计的组件,如非最大抑制过程或锚生成,这些组件显式地编码了我们对任务的先验知识。新框架的主要组

评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值