【论文笔记】TPH

【论文笔记】TPH

TPH-yolov5–基于transformer的改进yolov5的无人机目标检测

这里是原文~~原文

本文在YOLOv5的基础上加入了一些新的技术，比如ViT、CBAM和一些Tricks（数据增广、多尺度测试等），最终命名为TPH-YOLOv5的目标检测器，特别擅长在无人机的目标捕捉。
工作单位： 北京航空航天大学

TPH-yolov5整体架构：

本文的主要贡献：

1.在yolov5的基础上，又增加了一个预测头来检测不同尺度的物体。
2.用transformer prediction heads（TPH）替换原来的预测头部，以利用自注意力机制探索预测潜力。
3.集成了卷积块注意力模型（CBMA）以在具有密集对象的场景中找到注意力区域。
4.一些tricks：数据增强、多尺度检测、多模型集成和利用额外的分类器。

模型选择：