[人工智能]TPH-yolov5 小目标检测

原文链接：https://blog.csdn.net/stf1065716904/article/details/125487840

0. 序言

论文：《TPH-YOLOv5: Improved YOLOv5 Based on Transformer Prediction Head for Object Detection on Drone-captured Scenarios》

代码：GitHub - TfeiSong/tph-yolov5: CBAM-TPH-biFPN-yolov5
TPH：Transformer Prediction Head

总结比较实用的点：

• 1. tph（transformer prediction head）。能够捕获更多的全局信息和上下文信息（使用的self-attention和全连接）。
• 2. CBAM提高通道和空间注意力机制。简单有效好嵌入CNN架构。
• 3. 增加小目标预测head。在大尺寸featur

了解本专栏 订阅专栏 解锁全文 超级会员免费看